判断是不是树

判断是否为树的的依据可以简化为一句话: 是否有且仅有一个入度为0的节点。上只用映射就可以解决,但是要排除自环的情况。


#include
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;
}

你可能感兴趣的:(上机)