【luogu2700】逐個擊破

並查集,加最小生成樹技巧。循環搜索邊的最大權值,如果邊兩邊是需要隔離開的,則刪除該邊。
直接考慮原問題比較困難,我們可以這麼想:刪去的最少=留下來的最多。
那麼我們考慮用類似於最小生成樹的思想。在使用Kruskal算法時,並查集還要保存一個是否已經有敵人的城市。然後合併的時候必須要兩個集合不是都有敵人的城市(最多隻有一個集合有敵人的城市)纔可以合併。

#include<cstdio>
#define N 110000
struct node1{
	int x,y,z;
};
struct node2{
	bool f;int father;
};
node1 data[N];
node2 a[N];
int aa,n,k,h[N];
int gf(int x){
	if (a[x].father==x) return x;
	a[x].father=gf(a[x].father);
	return (a[x].father);
	
}
void qsort(int l,int r){
	int i=l,j=r,mid=(data[(l+r)/2].z);
	node1 t;
	while (i<=j){
		while (data[i].z>mid)++i;
		while (data[j].z<mid)--j;
		if (i<=j){
			t=data[i];data[i]=data[j];data[j]=t;
			++i;--j;
		}
	}
	if (l<j) qsort(l,j);
	if (i<r) qsort(i,r);
}
int main(){
	freopen("attack.in","r",stdin );
	freopen("attack.out","w",stdout);
	int num=0;
	scanf("%d%d",&n,&k);
	for (int i=0;i<n;++i){
		a[i].father=i;
		a[i].f=true;
	}
	for (int i=1;i<=k;++i) {
		scanf("%d",&aa);
		a[aa].f=false;
	}
	
	for (int i=1;i<n;++i){
		int a1,a2,a3;
		scanf("%d%d%d",&a1,&a2,&a3);
		data[++num].x=a1;
		data[num].y=a2;
		data[num].z=a3;
	}
	qsort(1,n-1);
	long long ans=0;
	for (int i=1;i<n;++i){
		int x=data[i].x,y=data[i].y;
		int fa=gf(x),fb=gf(y);
		if ((a[fa].f==a[fb].f)&&(a[fa].f==true)&&fa!=fb) {
			a[fa].father=a[fb].father;
			continue;
		}
		if (a[fa].f!=a[fb].f){
			a[fa].father=a[fb].father;
			a[fb].f=false;
			continue;
		}
		if ((a[fa].f==a[fb].f)&&(a[fa].f==false)){
			ans+=data[i].z;
		}
	}
	printf("%lld",ans);
	//for (int i=1;i<n;++i) printf("%d ",data[i].z);
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章