바이너리 트리의 루트 노드를 받고 해당 트리를 순회하며
노드의 값들을 2차원 벡터로 depth에 따라 지그재그 순서로 모아 반환해야 한다.
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>> result{};
if (root == nullptr)
{
return result;
}
SearchDFS(root, 0, result);
int length = result.size();
for (int i = 1; i < length; i += 2)
{
std::reverse(result[i].begin(), result[i].end());
}
return result;
}
private:
void SearchDFS(TreeNode* node, int index, vector<vector<int>>& result)
{
if (result.size() == index)
{
result.push_back({});
}
result[index].push_back(node->val);
if (node->left)
{
SearchDFS(node->left, index + 1, result);
}
if (node->right)
{
SearchDFS(node->right, index + 1, result);
}
}
};