二分木ジグザグレベルの順序トラバーサル
二分木が与えられた場合、そのノードの値のジグザグレベルの順序トラバーサルを返します。(つまり、左から右へ、次に右から左へと次のレベルへ、そして交互に)。例:与えられた二分木{3,9,20、#、#、15,7}、
3 / \ 9 20 / \ 15 7
ジグザグレベルの順序トラバーサルを次のように返します。
[ [3], [20,9], [15,7] ]
個人的には 、
時間計算量 = O(n *高さ)、nはノードの数、高さは与えられた二分木の高さだと思います。getHeight() => O(n) traverseSpecificLevel() => O(n) reverseVector() => O(n) swap() => O(1)
C ++
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
#include <vector>
using namespace std;
class Solution {
public:
vector<vector<int> > zigzagLevelOrder(TreeNode *root) {
vector<vector<int>> list;
// Input validation.
if (root == NULL) return list;
// Get the height of the binary tree.
int height = getHeight(root);
bool left_to_right = true;
for (int level = 0; level <= height; level ++) {
vector<int> subList;
traverseSpecificLevel(root, level, subList);
if (left_to_right == true) {
// Add subList into list.
list.push_back(subList);
// Update left_to_right flag.
left_to_right = false;
} else {
// Reverse subList.
reverseVector(subList);
// Add reversed subList into list.
list.push_back(subList);
// Update left_to_right flag.
left_to_right = true;
}
}
return list;
}
int getHeight(TreeNode *root) {
// Base case.
if (root == NULL || (root->left == NULL && root->right == NULL)) return 0;
else return 1 + max(getHeight(root->left), getHeight(root->right));
}
void traverseSpecificLevel(TreeNode *root, int level, vector<int> &subList) {
// Base case.
if (root == NULL) return;
if (level == 0) {
subList.push_back(root->val);
return;
}
// Do recursion.
traverseSpecificLevel(root->left, level - 1, subList);
traverseSpecificLevel(root->right, level - 1, subList);
}
void reverseVector(vector<int> &list) {
// Input validation.
if (list.size() <= 1) return;
int start = 0;
int end = list.size() - 1;
while (start < end) {
swap(list, start, end);
start ++;
end --;
}
}
void swap(vector<int> &list, int first, int second) {
int tmp = list[first];
list[first] = list[second];
list[second] = tmp;
}
};
あなたは線形時間でそれを行うことができます。サイズがmax_heightのベクトル>結果を作成します。ノードのレベルを維持しながら、ツリーを再帰的にトラバースします。すべてのノードについて、その値をresult [level]にプッシュバックします。結果[1]、結果[3]、...を逆にするだけではありません。
ちなみに、swap(x,y)
関数とreverse(a.begin(), a.end())
関数(ここa
で、はベクトル)があり、自分で実装する代わりにそれらを使用することができます。それを含めますalgorithm
。
この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。
侵害の場合は、連絡してください[email protected]
コメントを追加