洛谷P2619 [国家集训队2]Tree I 题解

Description

给你一个无向带权连通图，每条边是黑色或白色。让你求一棵最小权的恰好有need条白色边的生成树。
题目保证有解。

---

Input

第一行V,E,need分别表示点数，边数和需要的白色边数。
接下来E行
每行s,t,c,col表示这边的端点(点从0开始标号)，边权，颜色(0白色1黑色)。

---

Output

一行表示所求生成树的边权和。

---

Solution

奇妙的做法增加了！
一看 最小生成树！
二看 还有边数限制！

然后我就在想怎么把白色往前后搞
然后失败了

回想kruskal 是将边按边权从小到大排序
那么我们是否能通过修改边权以达到把白色往前后搞的操作呢
这就相当于我们得到了一个加了边权修改量的答案 减去修改量就是正确答案
那么我们二分一个边权修改量(正负均可) 能够达到need条白边就是答案
不过注意在排序时白边和黑边边权相等时选白边

---

Code

#include<bits/stdc++.h>
using namespace std;
struct data{
	int x,y,v,c;
}e[1000010];
int fa[1000010];
int sum,ans,tmp,n,m,k,cnt;
bool cmp(data a,data b){
	if(a.v==b.v) return a.c<b.c;
	else return a.v<b.v;
}
int find(int x){
	if(fa[x]==x) return x;
	return fa[x]=find(fa[x]);
}
void kruskal(){
	sort(e+1,e+m+1,cmp);
	for(int i=1;cnt!=n-1;i++){
		int xx=find(e[i].x),yy=find(e[i].y);
		if(xx==yy) continue;
		if(xx!=yy){
			cnt++;
			fa[xx]=yy;
			if(e[i].c==0) tmp++;
			sum+=e[i].v;
		} 
	}
}
int main(){
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=m;i++){
		scanf("%d%d%d%d",&e[i].x,&e[i].y,&e[i].v,&e[i].c);
		e[i].x++,e[i].y++;
	}
	int l=-100,r=100;
	while(l<=r){
		sum=cnt=tmp=0;
		int mid=(l+r)>>1;
		for(int i=1;i<=m;i++)
		if(!e[i].c) e[i].v+=mid;
		for(int i=1;i<=n;i++) fa[i]=i;
		kruskal();
		if(tmp>=k){
			l=mid+1;
			ans=sum-k*mid;
		}
		else r=mid-1;
		for(int i=1;i<=m;i++)
		if(!e[i].c) e[i].v-=mid;
	}
	printf("%d\n",ans);
}