Week7 A - TT 的魔法猫 [HDU-1704]

原题链接

http://acm.hdu.edu.cn/showproblem.php?pid=1704

题目大意

比赛，N个人，M个胜负关系。每个胜负关系为 A B，表示 A 能胜过 B，且胜负关系具有传递性。即 A 胜过 B，B 胜过 C，则 A 也能胜过 C。

输入

第一行给出数据组数。
每组数据第一行给出 N 和 M（N , M <= 500）。
接下来 M 行，每行给出 A B，表示 A 可以胜过 B。

输出

对于每一组数据，判断有多少场比赛的胜负不能预先得知。注意 (a, b) 与 (b, a) 等价，即每一个二元组只被计算一次。

基本思路

我们需要统计多少场比赛的胜负关系无法得知，也就是说，要利用传递性，把所有可以得出胜负关系的比赛找出。

一个很好的办法是使用Flody算法。虽然Flody算法是用来求最短路的，但实际上，这道题和求最短路异曲同工，Flody同样可以很好的解决。如果我们把选手看作是图上的节点，选手A可以胜过选手B看做AB间存在通路，使用Flody后，我们就可以得到任意两点之间是否存在通路(两点之间距离是否为无穷)。

由于胜负关系只有两种状态，可以把所有的胜负关系用一个bool类型二维数组cmp[N][N]来表示(最初数组的所有元素初始化为false)，cmp[A][B]=true就表示A可以胜过B，反之表示A不能胜过B。我们可以将Flody算法稍作变形，并进行一定的剪枝处理，代码如下：

void flody()
{
	for(int k=1;k<=N;k++)
	{
		for(int i=1;i<=N;i++)
		{
			if(!cmp[i][k])	continue;
			for(int j=1;j<=N;j++)
			{
				cmp[i][j] = cmp[i][j] || (cmp[i][k] && cmp[k][j]);
			}
		}
	}
}

最后，我们只需要统计满足$cmp[i][j]=cmp[j][i]=false(1<=i<j<=N)$的(i,j)数量，即是无法确定胜负的比赛数量。

完整代码

#include<iostream>
using namespace std;
int T, N, M, A, B;
bool cmp[505][505];
void flody()
{
	for(int k=1;k<=N;k++)
	{
		for(int i=1;i<=N;i++)
		{
			if(!cmp[i][k])	continue;
			for(int j=1;j<=N;j++)
			{
				cmp[i][j] = cmp[i][j] || (cmp[i][k] && cmp[k][j]);
			}
		}
	}
}

int getRes()
{
	int res=0;
	for(int i=1;i<N;i++)
	{
		for(int j=i+1;j<=N;j++)
		{
			if(cmp[i][j] == cmp[j][i])	res++;	
		}
	}
	return res;
}
int main ()
{
	cin>>T;
	for(int w=0;w<T;w++)
	{
		cin>>N>>M;
		for(int i=1;i<=N;i++)
		{
			for(int j=1;j<=N;j++)
			{
				cmp[i][j]=false;
			}
		}
		for(int i=0;i<M;i++)
		{
			cin>>A>>B;
			cmp[A][B]=true;
		}
		flody();
		cout<<getRes()<<endl;
	}
	return 0;
}