Populating Next Right Pointers in Each Node II

来源:互联网 发布:linux启动tomcat日志 编辑:程序博客网 时间:2024/04/29 16:28

Follow up for problem "Populating Next Right Pointers in Each Node".

What if the given tree could be any binary tree? Would your previous solution still work?

Note:

  • You may only use constant extra space.

For example,
Given the following binary tree,

         1       /  \      2    3     / \    \    4   5    7

After calling your function, the tree should look like:

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

/** * Definition for binary tree with next pointer. * struct TreeLinkNode { *  int val; *  TreeLinkNode *left, *right, *next; *  TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {} * }; */class Solution {public:    void connect(TreeLinkNode *root) {        if(root==NULL) return;        outside(root->left,root->right);        connect(root->left);        connect(root->right);    }    void outside(TreeLinkNode* l,TreeLinkNode* r){        if(l==NULL||r==NULL) return;        else l->next=r;        outside(l->left,r->right);        outside(l->left,r->left);        outside(l->right,r->right);        outside(l->right,r->left);       }};


0 0