前言
POJ 判断是否是树的一道题目。
题目概述
AC代码
#include<iostream>
#include<cstring>
using namespace std;
const int N = 1e5 + 10;
int ne[N], e[N], h[N], idx;
int ent[N], d[N], has[N], p[N], q[N];
int hh = 0, tt = -1, k;
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx++, ++ent[b];
}
void init()
{
memset(has, 0, sizeof has);
memset(p, 0, sizeof p);
memset(ent, 0, sizeof ent);
memset(h, -1, sizeof h);
idx = 0, hh = 0, tt = -1, k = 0;
}
void bfs(int root)
{
memset(d, -1, sizeof d);
q[++tt] = root;
d[root] = 0;
while(hh <= tt)
{
int u = q[hh++];
for(int i = h[u]; i != -1; i = ne[i])
{
int j = e[i];
if(d[j] == -1)
{
d[j] = d[u] + 1;
q[++tt] = j;
}
}
}
}
int main()
{
int a, b;
int id = 0;
while (1)
{
init();
bool tag = true;
bool blank = true;
while (1)
{
cin >> a >> b;
if (a == 0 && b == 0)
{
++id;
int root = -1;
for(int i = 0; i < k; ++i)
{
int j = p[i];
if(ent[j] == 0) //检查入度为0的根节点是否只有一个
{
if(root != -1) tag = false;
root = j;
}
else if(ent[j] > 1) tag = false;
}
if(root == -1) tag = false; //没有找到根节点
else
{
bfs(root);
for(int i = 0; i < k; ++i)
{
int j = p[i];
if(d[j] == -1) tag = false; //如果有从根节点无法到达的点
}
}
if(tag || blank) printf("Case %d is a tree.\n", id);
else printf("Case %d is not a tree.\n", id);
break;
}
else if (a == -1 && b == -1) return 0;
blank = false;
if(a == b) tag = false;
add(a, b);
if(!has[a]) p[k++] = a;
if(!has[b]) p[k++] = b;
++has[a], ++has[b];
}
}
}
分析思路
1.判断是否是树的方法题目中已经给出:第一,有且只有一个根节点,即入度为0的点有且只有一个。第二,除了根节点外的每个节点有且只有一条边指向它。那么就是除了根节点外的所有点的入度都为1。第三,从根到每个节点都有唯一有向边序列。在第二点条件中我们已经能够保证有向边序列不会超过1,那么如何保证存在性呢?可以在找到根节点后跑一遍bfs更新各点离根节点的距离,如果这张图各点不是都从根可达的,那么无法使得所有点的距离都被成功更新。
2.这题还有一点小坑的地方,那就是当一组案例为空的时候,也算作是一颗树,要特判一下。
文末广告
学习算法和数据结构真的是个很累的过程,不会做只能求助于题解。 因为写代码这个东西基本上是千人千面。同时网络上搜到的题解很多要么用到的是自己还没学到的知识,看不懂;要么内核过于简陋,只能糊弄当前题目,不具有普适性。
如果你是一个喜欢做洛谷,ACwing和PTA的题目的同学,欢迎关注我的博客,我主要在这三个平台上做题,认为有价值和有难度的题目我会写题解发布出来。
TreeTraverler的往期文章