[Ec Final 2018] Eventual … Journey

题目描述：

又N个车站，M条边，每个车站分为黑白两种
如果两个点颜色一样，可以花1的花费到达，或者两个点有边直接连接。
问每个点可以到达的点的总花费的最小值

题目分析：

互相到达的点花费只有三种可能
1：互相连接的点或者颜色一样的点
2：如果一个点链接着颜色不一样的点，我就可以通过这个点使用2的代价到达所有颜色不一样的点
3：如果一个点没有链接颜色不一样的点，去的点（即终点为颜色不一样的点也没有链接颜色不一样的点，这两个点之间就可以用3的花费到达

题目链接：

题目链接

AC代码：

#include <cstdio>
#include <algorithm>
const int maxm=1e5+100;
int val[maxm],ok[maxm],num[maxm];
int cnt[2],scnt[2];
int n,m;
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&val[i]);
		cnt[val[i]]++;
	}
	for(int i=1;i<=m;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		if(val[u]!=val[v]) num[u]++,num[v]++;
	}
	for(int i=1;i<=n;i++)
	 if(!num[i]) scnt[val[i]]++;
	for(int i=1;i<=n;i++)
	{
		int w1,w2,w3;
		w1=w2=w3=0;
		w1=cnt[val[i]]-1+num[i];
		if(num[i]) w2=cnt[!val[i]]-num[i];
		else w2=cnt[!val[i]]-scnt[!val[i]],w3=scnt[!val[i]];
		printf("%d ",w1+2*w2+3*w3);
	}
	return 0;
}