* 의사코드
핵심은 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;
}
};
'Algorithm > bfs' 카테고리의 다른 글
[Algo] 백준 벽부수고 이동하기 c++ // 최단거리는 bfs 3차원 (0) | 2024.09.13 |
---|---|
백준 16928 뱀과 사다리 게임 c++ // dp불가능한경우, bfs복습 (0) | 2024.05.13 |
백준 9205 맥주마시면서걸어가기 c++ // bfs, 출력시 "\n을 빼먹지마라.." (0) | 2024.03.09 |
백준 5014 스타트링크 c++ // bfs 를 사용하라. (dfs는 시간초과) (0) | 2024.03.04 |
백준 13549 숨바꼭질3 c++ // pq를 이용한 bfs 구현방법 (0) | 2023.11.29 |