두 갈래 나무 층계 반복 | 지그재그로 두 갈래 나무 인쇄
2257 단어 데이터 구조와 알고리즘
레이어별 인쇄
폭 우선 검색.
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
vector > Print(TreeNode* pRoot) {
vector> ret;
if (pRoot == NULL) {
return ret;
}
queue q;
q.push(pRoot);
while (!q.empty()) {
int len = q.size();
vector tmp(len);
for (int i = 0; i < len; ++i) {
TreeNode* cur = q.front(); q.pop();
tmp[i] = cur->val;
if (cur->left) q.push(cur->left);
if (cur->right) q.push(cur->right);
}
ret.push_back(tmp);
}
return ret;
}
};
지그재그 인쇄
BFS를 수정합니다.
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
vector > Print(TreeNode* pRoot) {
vector> ret;
if (pRoot == NULL) {
return ret;
}
vector q;
q.push_back(pRoot);
bool sign = true; // sign true ,
for (int i = 0; i < q.size(); ) {
int start = i, end = q.size() - 1;
vector tmp(end-start+1);
int k = 0;
for (int j = start; j <= end; ++j) {
tmp[k++] = q[j]->val;
if (q[j]->left) q.push_back(q[j]->left);
if (q[j]->right) q.push_back(q[j]->right);
}
if (!sign) std::reverse(tmp.begin(), tmp.end());
ret.push_back(tmp);
i = end + 1;
sign = !sign;
}
return ret;
}
};
다음으로 전송:https://www.cnblogs.com/ilovezyg/p/7591027.html
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
두 갈래 나무의 깊이가 두루 다니다텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.