宽度优先搜索套路

相比于我熟悉的深度优先搜索策略，宽度优先搜索代码如下。 这是层次遍历代码。

/** * 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);