将预先订购的前任与其后继者连接到二叉搜索树

时间:2015-12-30 16:57:56

标签: c tree

实际上在一次采访中我被要求编写一个代码,通过该代码,二进制搜索树中的每个节点都有一个额外的指针即“下一个”我们必须将每个节点的这个指针连接到它的预订后继者,可以任何一个建议我代码,因为我无法这样做。树节点具有以上结构: -

 struct node {
        int data ;
        struct node *left,*right;
        struct node *next; //this pointer should point to pre order successor 
        };
谢谢。

感谢破解解决方案,下面是用c: -

编写的整个代码
#include <stdio.h>
#include <stdlib.h>

struct node
{
        int data;
        struct node *left,*right,*next;
};

struct node* getNode(int data)
{
        struct node* temp=(struct node*)malloc(sizeof(struct node));
        temp->left=temp->right=NULL;
        temp->data=data;
        return temp;
}

void insert(struct node**root,int data)
{
    if(!*root)
    {
        *root=(struct node*)malloc(sizeof(struct node));
        (*root)->left=(*root)->right=NULL;
        (*root)->data=data;
    }
    else if(data<(*root)->data)
        insert(&((*root)->left),data);
    else if(data>(*root)->data)
        insert(&((*root)->right),data);
}

struct node* preorderSuccessor(struct node* root,struct node* p)
{
    int top,i;
    struct node *arr[20];
    if(!root || !p)
        return NULL;
    if(p->left)
        return p->left;
    if(p->right)
        return p->right;
    top=-1;
    while(root->data!=p->data)
    {
        arr[++top]=root;
        if(p->data<root->data)
            root=root->left;
        else
            root=root->right;
    }
    for(i=top;i>=0;i--)
    {
        if(arr[i]->right)
        {
            if(p!=arr[i]->right)
                return arr[i]->right;
        }
        p=arr[i];
    }
    return NULL;

}

void connect(struct node* parent,struct node *r)
{
        if(r)
        {       connect(parent ,r->left);    
                r->next = preorderSuccessor(parent,r);
                connect(parent,r->right);
        }
}


int main()
{
        struct node* root=NULL,*temp=NULL;
        int arr[]={10,11,2,3,9,8,4,5},size,i;
        size=sizeof(arr)/sizeof(arr[0]);

        for(i=0;i<size;i++) 
          insert(&root,arr[i]);
        connect(root,root);
        struct node *ptr = root;
        while(ptr){
          // -1 is printed if there is no successor
          printf("Next of %d is %d \n", ptr->data, ptr->next? ptr->next->data: -1);
          ptr = ptr->next;
        }


        return 0;
}

1 个答案:

答案 0 :(得分:0)

正如尤金所说:因此,通过前序遍历和连接来遍历树。为此,您需要知道最后访问过的节点(如果有的话)。

通过将引用传递给前一个节点,您可以使用通常的递归方法。这必须是在整个递归期间有效的变量的地址,因为前一个节点不一定更接近根。您可以使用全局变量,但在包装函数中创建的变量可能更好:

void connect_r(struct node *node, struct node **whence)
{
    if (node) {
        if (*whence) (*whence)->next = node;
        *whence = node;

        connect_r(node->left, whence);
        connect_r(node->right, whence);
    }
}

void connect(struct node *head)
{
    struct node *p = NULL;

    connect_r(head, &p);
    if (p) p->next = NULL;
}

p中的指针connect,其地址传递给递归函数connect_r,保存了下一个应更新next指针的节点。更新不会发生在第一个访问节点上。并且在递归之后,必须将最后访问过的节点的next成员显式设置为NULL

或者,您可以使用堆栈迭代地连接节点:

void connect(struct node *head)
{
    struct node *p = NULL;
    struct node **prev = &p;

    struct node *stack[32];    // To do: Prevent overflow
    size_t nstack = 0;

    if (head) stack[nstack++] = head;

    while (nstack) {
        struct node *node = stack[--nstack];

        *prev = node;
        prev = &node->next;

        if (node->right) stack[nstack++] = node->right;
        if (node->left) stack[nstack++] = node->left;
    }

    *prev = NULL;
}

连接的next指针是当前树的快照。节点的插入,删除和重新排列将导致next链无效。 (但只要树的左右链接一致,它就可以通过调用connect再次生效。)