POJ Is It A Tree？ （bfs，图论）

前言

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的往期文章