Mini

[알고리즘] 리트코드 116. 각 노드의 다음 오른쪽 포인터 채우기 c++ // bfs 트리순회 본문

Algorithm/bfs

[알고리즘] 리트코드 116. 각 노드의 다음 오른쪽 포인터 채우기 c++ // bfs 트리순회

Mini_96 2024. 7. 1. 16:06

* 의사코드

핵심은 rightNode에 직전우측노드를 캐시해놓고

next를 캐시된 그걸로 대입해주는것이다.

레벨별로 (rightNode의 초기값은 nullptr 이다.)

 

 

* 전체코드

class Solution {
public:
    Node* connect(Node* root) {
        if(!root) return nullptr; //null check 후에 root->next 등에 접근가능함
        queue<Node*> q;
        q.push(root);
        while(q.size()){
            Node* rightNode = nullptr;
            for(int i=q.size();i;i--){ //레벨별 구분위한 반복문(1,2,4...)
                auto cur = q.front(); q.pop();
                cur->next = rightNode; 
                rightNode=cur;  //직전우측노드를 캐시!
                if(cur->right){ //밑에자식이있는경우 추가
                    q.push(cur->right);
                    q.push(cur->left);
                }
            }
        }
        return root;
    }
};