宽度优先搜索套路

相比于我熟悉的深度优先搜索策略，宽度优先搜索代码如下。

这是层次遍历代码。

/**
* 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<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> result;
if(!root) return result;
queue<TreeNode*> q;
q.push(root);
while(!q.empty()){
vector<int> level;
int size=q.size();
for(int i=0;i<size;i++){
TreeNode* t=q.front();
q.pop();
level.push_back(t->val);
if(t->left){
q.push(t->left);
}
if(t->right){
q.push(t->right);
}
}
result.push_back(level);

}
return result;

}
};

上面是层次遍历代码，下面是把bfs结果都存进一个vector里。就改被注释掉的代码，再加一行就可以了，基本没变。

/**
* 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<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> result;
if(!root) return result;
queue<TreeNode*> q;
q.push(root);
while(!q.empty()){
//vector<int> level;
int size=q.size();
for(int i=0;i<size;i++){
TreeNode* t=q.front();
q.pop();
//level.push_back(t->val);
result.push_back(t->val);
if(t->left){
q.push(t->left);
}
if(t->right){
q.push(t->right);
}
}
//result.push_back(level);

}
return result;

}
};

下是用dfs实现层次遍历

vector<vector<int>> result;
vector<vector<int>> levelOrderBottom(TreeNode* root) {
if(!root) return result_final;
bfs(root,0);
return result;
}
void bfs(TreeNode* root,int level)
{
if(root==NULL)
return;
if(result.size()<level+1)
result.push_back(vector<int>{});
result[level].push_back(root->val);
bfs(root->left,level+1);
bfs(root->right,level+1);
}

其中，最重要的是传递了level这个参数，然后让每个节点在对应的vector中存储

就是这一句很重要

result[level].push_back(root->val);