力扣103 二叉树的锯齿形层序遍历

最新推荐文章于 2024-11-13 11:37:20 发布

火龙果573

最新推荐文章于 2024-11-13 11:37:20 发布

阅读量183

点赞数 3

文章标签： leetcode 深度优先算法数据结构

本文链接：https://blog.csdn.net/weixin_72777916/article/details/139893495

版权

/**
 * 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:
    void dfs(TreeNode *root, int k, vector<vector<int>> &ans) {
        if (root == NULL) return ; // 当前节点为空时结束
        if (k == ans.size()) ans.push_back(vector<int>()); // k ==ans.size() 表示没有当前层需要先添加一个进去
        ans[k].push_back(root->val); // 将第k层的值赋到ans中
        dfs(root->left, k + 1, ans);// 遍历左子树
        dfs(root->right, k + 1, ans); // 遍历右子树
        return ;
    }
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int>> ans; // 答案保存在了ans
        dfs(root, 0, ans);//深度优先遍历 传入根节点 root 当前层数 0 以及要保存到的二维数组 ans
        for (int i = 0; i < ans.size(); ++i) {
            if (i & 1) { // 偶数层不变  奇数层进行倒序输出
                for (int j = 0; j < ans[i].size() / 2; ++j) {
                    swap(ans[i][j], ans[i][ans[i].size() - 1 - j]); // 第一个和最后一个交换第二个和倒数第二个交换以此类推
                }
            }
        }
        return ans;
    }
};