用 C++ 从给定的前序遍历构造 BST - 设置 1
假设我们有一个前序遍历。从这个遍历中。我们必须生成树,所以如果遍历像 [10, 5, 1, 7, 40, 50],则树将像 −
要解决此问题,我们将使用此技巧。该技巧是为每个节点设置一个范围 {min… max}。首先,我们将范围初始化为 {INT_MIN… INT_MAX}。第一个节点肯定在范围内,所以在那之后我们将创建根节点。要构造左子树,请将范围设置为 {INT_MIN… root->data}。如果一个值在范围内 {INT_MIN… root->data},则该值是左子树的一部分。要构造右子树,请将范围设置为 {root->data… max… INT_MAX}。
示例
#include <iostream> using namespace std; class node { public: int data; node *left; node *right; }; node* getNode (int data) { node* temp = new node(); temp->data = data; temp->left = temp->right = NULL; return temp; } node* makeTreeUtil( int pre[], int* preord_index, int key, int min, int max, int size ) { if( *preord_index >= size ) return NULL; node* root = NULL; if( key > min && key < max ){ root = getNode( key ); *preord_index += 1; if (*preord_index < size){ root->left = makeTreeUtil( pre, preord_index, pre[*preord_index], min, key, size ); root->right = makeTreeUtil( pre, preord_index, pre[*preord_index],key, max, size ); } } return root; } node *makeTree (int pre[], int size) { int preord_index = 0; return makeTreeUtil( pre, &preord_index, pre[0], INT_MIN, INT_MAX, size ); } void inord (node* node) { if (node == NULL) return; inord(node->left); cout << node->data << " "; inord(node->right); } int main () { int pre[] = {10, 5, 1, 7, 40, 50}; int size = sizeof( pre ) / sizeof( pre[0] ); node *root = makeTree(pre, size); cout << "Inorder traversal: "; inord(root); }
输出
Inorder traversal: 1 5 7 10 40 50
广告