LeetCode - Binary Tree Level Order Traversal

最新推荐文章于 2024-02-06 03:28:26 发布

原创最新推荐文章于 2024-02-06 03:28:26 发布 · 515 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM

本文介绍了一种使用DFSID实现二叉树层次遍历的方法，并提供了详细的C++代码实现。通过递归地访问每个节点并按层级记录其值，最终形成一个二维数组，展示每层节点的值。

题意

输出每一层结点的值。

思路

一星期没做题了。。哎。。

我用的DFSID。

代码

#include <cstdio>
#include <vector>
using namespace std;
class Solution {
    int depth;
    vector<vector<int> >ans;
    bool flag;
public:
    void DFSID(TreeNode *curNode, int curDep)
    {
        if (curDep > depth) return;
        if (curDep < depth)
        {
            if (curNode->left) DFSID(curNode->left, curDep + 1);
            if (curNode->right) DFSID(curNode->right, curDep + 1);
        }
        if (curDep == depth)
        {
            ans[depth].push_back(curNode->val);
            flag = true;
        }
    }
    vector<vector<int> > levelOrder(TreeNode *root) {
        if (root == 0) return ans;
        int i, j;
        depth = -1;
        flag = true;
        while (flag)
        {
            depth++;
            flag = false;
            vector<int> tmpVec;
            ans.push_back(tmpVec);
            DFSID(root, 0);
        }
        ans.pop_back();
        return ans;
    }
};