hdu1325

hdu1325

Is It a tree

0x00 Tags

并查集

0x01 题目简介

判断有向图是否为树：

1. 每个点的入度都不大于1
2. 已经标记的点树根是自己的数量只能等于1，不然就是森林
3. 不能成环
4. 0 0也是树

此题和hdu1272类似，只是把无向边改成了有向边。多加一个判断条件，保证子节点是根节点。

0x02 代码

#include<bits/stdc++.h>

using namespace std;

const int maxn = 10010;

// in表示节点的入度
int root[maxn], sign[maxn], in[maxn];

// 
int k;
int flag;

void Init()
{
	for (int i = 1; i < maxn; i++)
	{
		root[i] = i;
		sign[i] = 0;
		in[i] = 0;
	}
}

int Find(int x)
{
	while (x != root[x])
	{
		x = root[x];
	}

	return x;
}


void Union(int x, int y)
{
	int fx = Find(x);
	int fy = Find(y);

	if (fx != fy)
	{
		root[fx] = fy;
		sign[fx] = 0;
		sign[fy] = 1;
	}
}


int main()
{
	int m, n;
	int cnt = 1;

	while (scanf("%d%d", &m, &n) != EOF)
	{
		if (m < 0 && n < 0) break;

		if (m == 0 && n == 0)
		{
			printf("Case %d is a tree.\n", cnt++);
			continue;
		}

		Init();

		k = 0;
		flag = true;

		Union(m, n);

		in[n]++;

		while (scanf("%d%d", &m, &n) != EOF&&(m+n))
		{
			Union(m, n);
			in[n]++;
		}

		for (int i = 1; i < maxn; i++)
		{
			if (in[i] > 1)
			{
				flag = false;
				break;
			}

			if (sign[i]) k++;

		}

		if (k > 1) flag = false;
		
		if (flag) printf("Case %d is a tree.\n", cnt++);
		else printf("Case %d is not a tree.\n", cnt++);


	}
}