leetcode@ [116/117] Populating Next Right Pointers in Each Node I & II (Tree, BFS)
Follow up for problem "Populating Next Right Pointers in Each Node".
What if the given tree could be any binary tree? Would your previous solution still work?
Note:
For example, Given the following binary tree,
1
/ \
2 3
/ \ \
4 5 7
After calling your function, the tree should look like:
1 -> NULL
/ \
2 -> 3 -> NULL
/ \ \
4-> 5 -> 7 -> NULL
/**
* Definition for binary tree with next pointer.
* struct TreeLinkNode {
* int val;
* TreeLinkNode *left, *right, *next;
* TreeLinkNode(int x) : val(x), left(NULL), right(NULL), next(NULL) {}
* };
*/
class node {
public:
TreeLinkNode* nd;
int lv;
node(TreeLinkNode* rhs, int l): nd(rhs), lv(l) {}
};
class Solution {
public:
void connect(TreeLinkNode *root) {
if(!root) return;
queue<node> q;
q.push(node(root, 0));
vector<node> vec;
while(!q.empty()) {
node top = q.front();
vec.push_back(top);
q.pop();
int cur_lv = top.lv;
TreeLinkNode* cur_nd = top.nd;
if(cur_nd->left) q.push(node(cur_nd->left, cur_lv+1));
if(cur_nd->right) q.push(node(cur_nd->right, cur_lv+1));
}
int l = 0, r = 1;
while(r < vec.size()) {
vec[l].nd->next = NULL;
while(r < vec.size() && vec[r].lv == vec[l].lv) {
vec[l].nd->next = vec[r].nd;
++l;
++r;
}
l = r;
r = l+1;
}
vec[vec.size()-1].nd->next = NULL;
}
};
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
다양한 언어의 JSONJSON은 Javascript 표기법을 사용하여 데이터 구조를 레이아웃하는 데이터 형식입니다. 그러나 Javascript가 코드에서 이러한 구조를 나타낼 수 있는 유일한 언어는 아닙니다. 저는 일반적으로 '객체'{}...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.