단순 반복 + 레이어 반복 (레이어 기록) 116.각 노드의 다음 오른쪽 노드 포인터 117을 채웁니다.각 노드의 다음 오른쪽 노드 포인터 채우기 II
/*
// Definition for a Node.
class Node {
public:
int val;
Node* left;
Node* right;
Node* next;
Node() : val(0), left(NULL), right(NULL), next(NULL) {}
Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}
Node(int _val, Node* _left, Node* _right, Node* _next)
: val(_val), left(_left), right(_right), next(_next) {}
};
*/
class Solution {
public:
Node* connect(Node* root) {
if(!root) return root;
root->next=NULL;
getnext(root->left,root->right);
return root;
}
private:
void getnext(Node* a,Node * b){
if(!a||!b) return;
a->next=b;
getnext(a->left,a->right);
getnext(a->right,b->left);
getnext(b->left,b->right);
return;
}
};
층차 반복 노드 채우기 (1) 각 층의 꼬리 바늘을 기록하고 층수를 기록하는 층차 반복
/*
// Definition for a Node.
class Node {
public:
int val;
Node* left;
Node* right;
Node* next;
Node() : val(0), left(NULL), right(NULL), next(NULL) {}
Node(int _val) : val(_val), left(NULL), right(NULL), next(NULL) {}
Node(int _val, Node* _left, Node* _right, Node* _next)
: val(_val), left(_left), right(_right), next(_next) {}
};
*/
class Solution {
public:
Node* connect(Node* root) {
if(!root) return root;
Q.push(root);
Node * end=root;
Node * cur;
Node * pre=end;
while(!Q.empty()){
Node* tmp=Q.front();
Q.pop();
if(tmp->left){
Q.push(tmp->left);
cur=tmp->left;
}
if(tmp->right){
Q.push(tmp->right);
cur=tmp->right;
}
if(tmp==end) {
if(pre)
pre->next=tmp;
tmp->next=NULL;
end=cur;
pre=NULL;
}
else {
if(pre)
pre->next=tmp;
pre=tmp;
}
}
return root;
}
private:
queue<Node* > Q;
};
(2) 각 층의 크기를 기록하여 층수를 기록하는 층차 반복
class Solution {
public:
Node* connect(Node* root) {
if(!root) return root;
Q.push(root);
Node *tmp;
Node *pre=NULL;
while(!Q.empty()){
int size=Q.size();
while(size--){
tmp=Q.front();
Q.pop();
if(pre) pre->next=tmp;
if(!size) tmp->next=NULL;
if(tmp->left)
Q.push(tmp->left);
if(tmp->right)
Q.push(tmp->right);
pre=tmp;
}
pre=NULL;
}
return root;
}
private:
queue<Node* > Q;
};
최적화: Q.front()를 이용하여pre 변수를 절약한다.
class Solution {
public:
Node* connect(Node* root) {
if(!root) return root;
Q.push(root);
Node *tmp;
while(!Q.empty()){
int size=Q.size();
while(size--){
tmp=Q.front();
Q.pop();
if(size) tmp->next=Q.front();
else tmp->next=NULL;
if(tmp->left) Q.push(tmp->left);
if(tmp->right) Q.push(tmp->right);
}
}
return root;
}
private:
queue<Node* > Q;
};
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
0부터 시작하는 LeetCode Day8 「1302. Deepest Leaves Sum」해외에서는 엔지니어의 면접에 있어서 코딩 테스트라고 하는 것이 행해지는 것 같고, 많은 경우, 특정의 함수나 클래스를 주제에 따라 실장한다고 하는 것이 메인이다. 빠른 이야기가 본고장에서도 행해지고 있는 것 같은 코...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.