pat-top 1001 Battle Over Cities - Hard Version

题目

  题意：给一张图，图上有好的路和坏的路，坏的路修复需要代价，将一个点移除，如果要保证其他点联通，有可能需要修复坏的路，问哪些点被移除产生的代价最大。

  题解：n次最小生成树，把好的路视为0花费的路加入就可以了。但是要注意，有可能修复所有的路仍然不连通，这种情况的花费应视为无限大。

代码：

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn = 510;
const int maxm = 150000;
const int inf = 0x3f3f3f3f;
int p[maxn];
int result[maxn];
struct node{
	int x,y,cost;
}e1[maxm],e0[maxm];
int cmp(node a,node b){
	return a.cost<b.cost;
}
int finde(int x){
	if(p[x]==x)return x;
	return p[x]=finde(p[x]);
}
int n,m;
int imax(int a,int b){
	return a>b?a:b;
}

int main()
{
	int i,j;
	int a,b,c,d;
	int num_e1=0,num_e0=0;
	scanf("%d%d",&n,&m);
	for(i=0;i<m;i++){
		scanf("%d%d%d%d",&a,&b,&c,&d);
		if(d){
			e1[num_e1].x=a;
			e1[num_e1].y=b;
			e1[num_e1++].cost=c;
		}
		else {
			e0[num_e0].x=a;
			e0[num_e0].y=b;
			e0[num_e0++].cost=c;
		}
	}
	//printf("nume0: %d nume1: %d\n",num_e0,num_e1);
	sort(e0,e0+num_e0,cmp);
	int num=0;
	int maxx=0;
	for(i=1;i<=n;i++){
		int sum=0;
		num=0;
		for(j=1;j<=n;j++)p[j]=j;
		for(j=0;j<num_e1;j++){
			if(e1[j].x==i||e1[j].y==i)continue;
			int xx,yy;
			xx=finde(e1[j].x);
			yy=finde(e1[j].y);
			if(xx==yy)continue;
			p[xx]=yy;
			//printf("add:%d %d\n",e1[j].x,e1[j].y);
			num++;
		}
		if(num==n-2){
			result[i]=0;
			continue;
		}
		for(j=0;j<num_e0;j++){
			if(e0[j].x==i||e0[j].y==i)continue;
			int xx,yy;
			xx=finde(e0[j].x);
			yy=finde(e0[j].y);
			if(xx==yy)continue;
			p[xx]=yy;
			sum+=e0[j].cost;
			//printf("add:%d\n",e0[j].cost);
			num++;
			if(num==n-2)break;
		}
		int rot=0;
		for(j=1;j<=n;j++){
			if(p[j]==j)rot++;
		}
		if(rot>2)sum=inf;
		result[i]=sum;
		//printf("%d:%d %d\n",i,sum,num);
	}
	for(i=1;i<=n;i++)maxx=imax(maxx,result[i]);
	if(maxx==0){
		printf("0\n");
		return 0;
	}
	bool fst=true;
	for(i=1;i<=n;i++){
		if(result[i]==maxx){
			if(fst)fst=false;
			else printf(" ");
			printf("%d",i);
		}
	}
	printf("\n");
	//printf("%d\n",maxx);
	//printf("%d",e0[0].cost);
    return 0;
}