leetcode BFS/DFS更新中

leetcode 102 二叉树层序遍历

bfs 模板

模板一

如果不需要确定当前遍历到了哪一层，BFS模板如下。
while queue 不空：
    cur = queue.pop()
    for 节点 in cur的所有相邻节点：
        if 该节点有效且未访问过：
            queue.push(该节点)

模板二

如果要确定当前遍历到了哪一层，BFS模板如下。
这里增加了level表示当前遍历到二叉树中的哪一层了，也可以理解为在一个图中，现在已经走了多少步了。size表示在当前遍历层有多少个元素，也就是队列中的元素数，我们把这些元素一次性遍历完，即把当前层的所有元素都向外走了一步。
level = 0
while queue 不空：
    size = queue.size()
    while (size --) {
        cur = queue.pop()
        for 节点 in cur的所有相邻节点：
            if 该节点有效且未被访问过：
                queue.push(该节点)
    }
    level ++;

题解代码

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        // bfs 模板 
        
        if(!root) return { };
        vector<vector<int>> res;
        queue<TreeNode *> que;
        que.push(root);
        while(!que.empty())
        {
            int ss=que.size();
            vector<int> temp;
            for(int i=0;i<ss;i++)
            {
                auto cur=que.top(); que.pop();
                if(cur->left) que.push(cur->left);
                if(cur-right) que.push(cur->right);
                temp.push_back(cue->val);
            }

            res.push_back(temp);
            temp.clear();
        }

        return
    }
};

dfs 题解记录

res.push_back(vector<int>());
这句代码是为了 提前申请空间
因为未分配的空间无法用 下标进行访问

class Solution {
public:

    vector<vector<int>> res;
    vector<vector<int>> levelOrder(TreeNode* root) {
        dfs(root,0);
        return res;
    }
    void dfs(TreeNode* root, int n)
    {
        if(!root) return ;
        if(n>=res.size())
            res.push_back(vector<int>());
        res[n].push_back(root->val);
        if(root->left) dfs(root->left,n+1);
        if(root->right) dfs(root->right,n+1);
    }
};