分类 广度优先搜索 下的文章

二叉树的中序遍历

问题描述

给定一个二叉树,返回它的中序 遍历。

示例:

输入: [1,null,2,3]
   1
    \
     2
    /
   3

输出: [1,3,2]

进阶: 递归算法很简单,你可以通过迭代算法完成吗?

思路

深度优先搜索

题解

/**
 * 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:
    vector<int> ret;
    vector<int> inorderTraversal(TreeNode* root) {
        dfs(root);
        return ret;
    }
    void dfs(TreeNode *node){
        if(node==nullptr){
            return ;
        }
        dfs(node->left);
        ret.push_back(node->val);
        dfs(node->right);
    }
};

二叉树的层次遍历

问题描述

给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

[
  [3],
  [9,20],
  [15,7]
]

思路

第一种方法:dfs
deep+1>ret.size(),在新的一层插入之前,ret.push_back(vector<int> ())都会申请一个新的vector<int>且为空。
利用deep深度,来控制每一层的元素在一个vector中。

题解

class Solution {
public:
    vector<vector<int>> ret;
    void dfs(int deep,TreeNode *cur){
        if(cur== nullptr) return;
        if(deep+1>ret.size()) ret.push_back(vector<int> ());
        ret[deep].push_back(cur->val);
        if(cur->left!= nullptr) dfs(deep+1,cur->left);
        if(cur->right!= nullptr) dfs(deep+1,cur->right);
    }
    vector<vector<int>> levelOrder(TreeNode *root) {
        dfs(0, root);
        return ret;
    }
};

思路

第二种方法:层次遍历
在访问过程中将同一层中的节点同时入队列。在将该queue中所有元素出队列的同时,将下一层的元素进队列,完成交接。

题解

class Solution {
public:
    vector<vector<int>> ret;
    vector<vector<int>> levelOrder(TreeNode *root) {
        if(root== nullptr) return ret;
        queue<TreeNode *> Q;
        Q.push(root);
        TreeNode* p;
        while (!Q.empty()){
            vector<int>a;
            int len=Q.size();
            for(int i=0;i<len;i++){
                p=Q.front();
                a.push_back(p->val);
                Q.pop();
                if(p->left!= nullptr) Q.push(p->left);
                if(p->right!= nullptr) Q.push(p->right);
            }
            ret.push_back(a);
        }
        return ret;
    }
};