leetcode之populating-next-right-pointers-in-each-node-ii

leetcode之populating-next-right-pointers-in-each-node-ii

题目

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,

After calling your function, the tree should look like:

题意

这道题目考察的是二叉树的层次遍历，给出一个二叉树，要求通过一个算法，可以让其每一层的节点通过next再形成一条链路。
该链表从左至右，最后一个是NULL。

解题思路

通过二叉树最经典的层次遍历去做，使用queue存储层次遍历的所有节点，然后循环判断，如果是该层循环中的最后一个几点，则直接next= NULL；
那么如何判断是否是该层的最后一个节点呢，则可以通过对应queue的size来获得，对应index = size-1的一定是当前这一层的最后一个节点。具体解题代码如下：

C++实现代码

/**
 * 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 == nullptr) return;
        queue<TreeLinkNode*> que;
        que.push(root);
        while(!que.empty()){
            int n=que.size();
            for(int i=0;i<n;i++){
                TreeLinkNode* tmp = que.front();
                que.pop();
                if(tmp->left){
                    que.push(tmp->left);
                }
                if(tmp->right){
                    que.push(tmp->right);
                }
                if(i !=n-1){
                    tmp->next =que.front();
                }else{
                    tmp->next = NULL;
                }
            }
        }
    }
};