6行代码AC LeetCode 116. Populating Next Right Pointers in Each Node

class Solution {
public:
    void connect(TreeLinkNode *root) {
        dfs(root);
    }
    
    void dfs(TreeLinkNode *root){
        if (root == NULL) return;
        if (root->left != NULL) root->left->next = root->right;
        if (root->next != NULL && root->right != NULL) root->right->next = root->next->left;
        if (root->next == NULL && root->right != NULL) root->right->next = NULL;
         
        dfs(root->left);
        dfs(root->right);
    }
};
已邀请:

要回复问题请先登录注册

返回顶部