给定一个有向图,需要我们判断其是否是一颗二叉树,如下,题目在验证二叉树
对于包含n个结点m条边无向图,如果他是一颗树,那么必须满足以下三个条件中的两个:
- m=n-1
- 该无向图连通
- 该无向图无环
对于连通性的判断,我们可以采用广度优先搜索,从根结点出发,将遍历到的结点放入一个set中,当一次广度优先搜索结束之后,set中的元素应该是所有元素的个数,如果存在在遍历过程中,访问到了已经放进set中的元素,说明有环存在。因为二叉树的遍历中,每个元素只会遍历一次,因此遍历到他之后可以直接将其放进set中。
同时可以通过统计入度的值,如果入度为0的元素的值多于1个,说明不是二叉树,如果存在入度的值大于1的情况,不是二叉树。代码如下
bool validateBinaryTreeNodes(int n, vector<int>& leftChild, vector<int>& rightChild) {
vector<int>degree(n);
for(int i=0;i<n;i++){ //统计入度的情况
if(leftChild[i]!=-1) degree[leftChild[i]]++;
if(rightChild[i]!=-1) degree[rightChild[i]]++;
}
int res=0;
int root=0;
for(int i=0;i<n;i++){
if(degree[i]==0) { //统计入度的结点
res++;
root=i;
}
if(degree[i]>1) return false;
}
if(res!=1) return false;
unordered_set<int>s; //统计遍历的情况
queue<int>q;
q.push(root);
s.insert(root);
while(!q.empty()){
auto cur=q.front();
q.pop();
if(leftChild[cur]!=-1){
if(s.count(leftChild[cur])) return false; //已经遍历过了,入度不为1
else {
s.insert(leftChild[cur]);
q.push(leftChild[cur]);
}
}
if(rightChild[cur]!=-1){
if(s.count(rightChild[cur])) return false;
else{
s.insert(rightChild[cur]);
q.push(rightChild[cur]);
}
}
}
return s.size()==n; //遍历的结果是所有结点都遍历了一次
}