每个节点的右向指针

给定一个二叉树

struct TreeLinkNode {
  TreeLinkNode *left;
  TreeLinkNode *right;
  TreeLinkNode *next;
}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL

初始状态下,所有 next 指针都被设置为 NULL

说明:

  • 你只能使用额外常数空间。
  • 使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
  • 你可以假设它是一个完美二叉树(即所有叶子节点都在同一层,每个父节点都有两个子节点)。

示例:

给定完美二叉树,

     1
   /  \
  2    3
 / \  / \
4  5  6  7

调用你的函数后,该完美二叉树变为:

     1 -> NULL
   /  \
  2 -> 3 -> NULL
 / \  / \
4->5->6->7 -> NULL

二叉树顺序遍历,多了个操作。要是所有题目都这么简单该多好啊。。。

 

/**
 * Definition for binary tree with next pointer.
 * struct TreeLinkNode {
 *  int val;
 *  struct TreeLinkNode *left, *right, *next;
 * };
 *
 */
void connect(struct TreeLinkNode *root) {
    if(!root)
        return ;
    struct TreeLinkNode **queue=(struct TreeLinkNode **)malloc(sizeof(struct TreeLinkNode *)*10000);
    struct TreeLinkNode *s;
    int front=-1,rear=-1;
    queue[++rear]=root;
    int count;
    while(front<rear)
    {
        count=rear-front;
        s=(struct TreeLinkNode *)malloc(sizeof(struct TreeLinkNode));
        while(count--)
        {
            s->next=queue[++front];
            s=queue[front];
            if(s->left)
                queue[++rear]=s->left;
            if(s->right)
                queue[++rear]=s->right;
        }
        s->next=NULL;
    }
}
posted @ 2018-07-17 16:44  onlyandonly  阅读(176)  评论(0编辑  收藏  举报