CareerCup-4.5

Write an algorithm to find the ‘next’ node (i e , in-order successor) of a given node in a binary search tree where each node has a link to its parent

#include <iostream>    
using namespace std;    
typedef int Data;    
    
struct Node    
{    
    Node(Data d):data(d){left = right = NULL;};    
    Data data;    
    Node* left;    
    Node* right;  
    Node* parent;   
};  
  
struct ListNode  
{  
    ListNode(Node* n):data(n){next = NULL;};  
    ListNode* next;  
    Node* data;  
};  
    
class Tree    
{    
public:    
    Tree():root(NULL){};    
    Node* root;    
    void insert(Data data, Node* node = NULL)    
    {    
        if(root == NULL)    
        {    
            root = new Node(data);  
            root->parent = NULL;  
            return;    
        }    
        if(node == NULL) node = root;    
        if(data == node->data)    
        {    
            return;    
        } else if(data < node->data) {    
            if(node->left == NULL)    
            {  
                node->left = new Node(data);  
                node->left->parent = node;        
            } else {  
                insert(data, node->left);    
            }  
        } else {    
            if(node->right == NULL)    
            {  
                node->right = new Node(data);   
                node->right->parent = node;   
            } else {  
                insert(data, node->right);    
            }  
        }    
    };  
    void findNext(Node* n)  
    {  
        if(n == NULL) return;  
        if(n->right != NULL)  
        {  
            Node *p = n->right;  
            while(p->left != NULL)  
                p = p->left;  
            cout<<p->data<<endl;  
        } else {  
            if(n->parent == NULL)  
                return;  
            if(n == n->parent->left)  
            {  
                cout<<n->parent->data<<endl;  
            } else {  
                Node* p = n;  
                while(p->parent != NULL && p == p->parent->right)  
                {  
                    p = p->parent;  
                }  
                if(p->parent != NULL)  
                    cout<<p->parent->data<<endl;  
            }  
        }  
    };  
};    
    
int main()    
{    
    Tree tree;    
    tree.insert(5);    
    tree.insert(1);    
    tree.insert(7);    
    tree.insert(4);    
    tree.insert(3);  
    Node *n = tree.root->left->right;  
    cout<<n->data<<endl;  
    tree.findNext(n);  
    system("pause");    
};

Comments

comments powered by Disqus