判断是否为树的的依据可以简化为一句话: 是否有且仅有一个入度为0的节点。上只用映射就可以解决,但是要排除自环的情况。
#include<bits/stdc++.h>
using namespace std;
map<int, int> inDegree;
bool isTree() {
if (inDegree.empty()) return true; //空树也算树
int rootCount = 0;
for (auto iter = inDegree.begin(); iter !=
inDegree.end(); iter++) {
if (iter->second == 0) rootCount++; //入度为0则计入根节点数
if (iter->second > 1 || rootCount > 1) {//若入度大于1或根节点数大于1,显然非树
return false;
}
}
return rootCount; //循环结束后,若根节点数非0,则返回true
}
int main() {
int a, b;
int caseNumber = 0;
bool wrong = false; //利用wrong布尔变量判断是否为树
while(scanf("%d%d", &a, &b) != EOF){
if (a == -1) break;
if (!a && !b) { //遇 0 0 输入,则判断是否为树,判断结束后重置变量
if (!isTree()) wrong = true;
if (!wrong) printf("Case %d is a tree.\n", ++caseNumber);
else printf("Case %d is not a tree.\n", ++caseNumber);
wrong = false; //变量、映射重置
inDegree.clear();
continue;
}
if (a == b) wrong = true; //排除自环,注意0 0的情况会在上一个if语句中判断
inDegree[a]; //比较随便的一种写法,这样写后,若inDegree[a]不存在,则会在映射中创造一个,并赋值为0
inDegree[b]++; //入度增加
}
return 0;
}