BFS
例子
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
/**
* 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:
int minDepth(TreeNode* root) {
if(root==NULL)
return 0;
//BFS initial
queue<TreeNode*> check;
check.push(root);
//不用看是否visited
int res=1;
while(!check.empty())
{
//判断是否到达底部
int sz=check.size();
for(int i=0;i<sz;i++)
{
TreeNode* a=check.front();
check.pop();
if(a->right==NULL&&a->left==NULL)
{
return res;
}
if(a->right!=NULL) check.push(a->right);
if(a->left!=NULL) check.push(a->left);
}
res++;
}
return res;
}
};
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val) {
val = _val;
}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public:
vector<vector<int>> levelOrder(Node* root) {
vector<vector<int>> res;
if(root==NULL)
return res;
queue<Node*> check;
check.push(root);//初始化
while(!check.empty())//循环条件,队列不为空
{
int sz=check.size();
vector<int> tem_res;
for(int i=0;i<sz;i++)//每次都先去除一层
{
Node* tem=check.front();
check.pop();
tem_res.push_back(tem->val);
for(int j=0;j<tem->children.size();j++)
{
check.push(tem->children[j]);//每去掉一个就加入相关的
}
}
res.push_back(tem_res);
}
return res;
}
};
都是一个套路,我们每次去掉一个,就加入他的相关子节点.是不是很简单