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;
}
};