BZOJ1116: [POI2008]CLO 并查集

1116: [POI2008]CLO

Time Limit: 10 Sec   Memory Limit: 162 MB

Submit: 866   Solved: 483

[Submit][Status][Discuss]

Description

Byteotia城市有n个 towns m条双向roads. 每条 road 连接 两个不同的 towns ,没有重复的road. 你要把其中一些road变成单向边使得：每个town都有且只有一个入度

Input

第一行输入n m.1 <= n<= 100000,1 <= m <= 200000 下面M行用于描述M条边.

Output

TAK或者NIE 常做POI的同学,应该知道这两个单词的了...

Sample Input

4 5
1 2
2 3
1 3
3 4
1 4

Sample Output

TAK

上图给出了一种连接方式.

题解：

n个点连边后会形成一些连通块，对于一个连通块：如果边数=点数-1，那么必然不能满足题意（因为一条边只能贡献给一个点），否则满足。所以对于每一个连通块想要满足题意的话，都要保证边数>=点数。（自己画一画更有助于理解）

既然维护连通块中的点数和边数，我们就可以用并查集了，对于每一个点我们维护它所在的连通块的根的信息，一共两个量：一个是连通块中的点数，一个是连通块中的边数。

每当加进来一条边（起点为f，终点为t）时，如果f，t在同一连通块中则该连通块的边数+1，点数不变；如果不在同一连通块中，点数直接变成两个连通块的和，边数为两个连通块之和+1；

最后把所有连通块查找一遍就好啦

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100005;
int n,m,fa[N],bel[N],w[N],d[N];
int find(int x)
{
	if(fa[x]==x) return x;
	fa[x]=find(fa[x]);
	return fa[x];
}
int main()
{
	scanf("%d%d",&n,&m);
	int x,y;
	for(int i=1;i<=n;i++) fa[i]=i,d[i]=1;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		int fx=find(x),fy=find(y);
		if(fx!=fy)
		{
			fa[fy]=fx;
			w[fx]+=w[fy]+1;
			d[fx]+=d[fy];
		}
		else w[fx]++;
	}
	for(int i=1;i<=n;i++)
	{
		x=find(i);
		if(w[x]==d[x]-1)
		{
			printf("NIE");
			return 0;
		}
	}
	printf("TAK");
}