LC 102 Binary Tree Level Order Traversal

基础中的基础,直接背就好了。用队列。因为需要在最后按层序打印,所以队列中记录层数。

AC代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    queue <std::pair<int, TreeNode *>> q;
    vector<vector<int>> levelOrder(TreeNode* root) {
        if (!root) return {};
        vector<vector<int>> ans;
        q.push({0, root});
        while (!q.empty()){
            auto tmp = q.front();
            q.pop();
            int level = tmp.first;
            TreeNode* c_root = tmp.second;
            if (ans.size() == level){
                ans.push_back(vector<int>{c_root->val});
            } else {
                ans[level].push_back(c_root->val);
            }
            if (c_root->left) q.push({level+1,c_root->left});
            if (c_root->right) q.push({level+1, c_root->right});
        }
        return ans;
    }
};

这种方法是纯粹按照“层序”遍历的。如果只是要求打印层间次序,也可以使用DFS,discussion中的代码:

public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        levelHelper(res, root, 0);
        return res;
    }
    
    public void levelHelper(List<List<Integer>> res, TreeNode root, int height) {
        if (root == null) return;
        if (height >= res.size()) {
            res.add(new LinkedList<Integer>());
        }
        res.get(height).add(root.val);
        levelHelper(res, root.left, height+1);
        levelHelper(res, root.right, height+1);
    }

事实上针对这道题目而言,DFS和“BFS”(第一种方法、层序遍历)而言是tradeoff

如果是完全二叉树(或者近似完全二叉树),则DFS的空间复杂度是O(logn),BFS是O(n)

如果是窄二叉树,则DFS的空间复杂度是O(N),BFS是O(1)


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值