LeetCode116,117. 每个节点的右向指针

code

// leetcode 116,117，每个节点的右向指针
// 这道题主要是层次遍历的应用，由于是完美的二叉树，所有有左子树的一定有右子树
// 同一个父节点，让左子树指向右子树就可以了。不同的父节点，则右子树指向同一层的左子树
// Definition
struct Node {
	int val;
	Node *left;
	Node *right;
	Node *next;
	Node() {};
	// 初始化
	Node(int _val Node* _left, Node* _right, Node* _next) {
		val = _val;
		left = _left;
		right = _right;
		next = _next;
	}
};
// 递归的解法
class Solution {
public:
    Node* connect(Node* root) {
        if(!root)    return NULL;
        // 如果左节点存在，则右节点一定存在
        if(root->left)  root->left->next = root->right;
        // 指向上一层的
        if(root->right) {
            if(root->next)
                root->right->next = root->next->left;
            else
                root->right->next = NULL;
        }
        connect(root->left);
        connect(root->right);
        return root;
    }
};
// 非递归的解法
class Solution {
public:
    Node* connect(Node* root) {
        if(!root)  return NULL;
        queue<Node*> q;
        q.push(root);  //一般如果使用队列/栈的话，最外层的判断条件都是 队列/栈 是否为空
        while(!q.empty()) {
            //层次遍历的非递归方法一般是一层一层地处理，所以需要用for循环
            for(int i = 0; i<q.size(); ++i) {
                Node* t = q.front(); q.pop();
                // 将该层当前节点弹出来后，指向队列中下一个节点（也即是同一层的右边节点）
                if(i < q.size()-1)  // 每一层最右边的节点指向空，不定义就是指向空
                    t->next = q.front();
                if(t->left)  q.push(t->left);
                if(t->right)  q.push(t->right);
            } 
        }
        return root;  //因为队列中存放的第一个是根节点，所以可以直接返回根元素
    }
};
// 117的题不是 完美的二叉树，递归的方法有些小复杂，非递归的方式还是可以用同样的方式