Codeforces 125E MST Company【k度最小生成树】

题目描述：

n个点m条无向边，在点1的度数恰为k的限制下求任意一棵最小生成树，输出边的编号，无解输出-1。
n<=5000,m<=100000

题目分析：

WQS二分。
给每条与1相连的边加上mid的权值后做最小生成树，如果选出的1边恰为k条，则符合条件，实际总权值为现权-k*mid。此时选出的树一定是在k限制下的最小生成树。如果小于k条，则加权要减小，r=mid-1；否则 l=mid；

二分可用整数，但是需要明确在边权相等时优先选1边（或者其他边，二分时l,r的变化由此决定），这样二分出的结果最后可能选出大于k条1边，但是我们强制选k条1边，得到的值同样是最小权值，因为这时多余的1边会换成边权相等的其他边。

注意对无解的判断，可能原图在1边全选时可以联通，但恰选k条时无法联通。

Code：

#include<bits/stdc++.h>
#define maxn 5005
#define maxm 100005
using namespace std;
int n,m,k,f[maxn],ans,num,cnt,t[maxn];
struct node{
	int x,y,w,p,id;
	bool operator < (const node &B)const{return p==B.p?x<B.x:p<B.p;}
}e[maxm];
int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
void Kruskal(int mid){
	for(int i=1;i<=m;i++) e[i].p=e[i].w+(e[i].x==1?mid:0);
	sort(e+1,e+1+m);
	for(int i=1;i<=n;i++) f[i]=i;
	ans=num=cnt=t[0]=0;
	for(int i=1,x,y;i<=m&&cnt<n-1;i++)
		if((x=find(e[i].x))!=(y=find(e[i].y))&&num+(e[i].x==1)<=k)
			f[x]=y,ans+=e[i].p,num+=(e[i].x==1),cnt++,t[++t[0]]=e[i].id;
}
int main()
{
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].w),e[i].id=i;
		if(e[i].x>e[i].y) swap(e[i].x,e[i].y);
	}
	Kruskal(-100000);
	if(cnt<n-1||num<k) return puts("-1"),0;
	int l=-100000,r=100000,mid;
	while(l<r){
		Kruskal(mid=(l+r+1)>>1);
		if(num>=k) l=mid;
		else r=mid-1;
	}
	Kruskal(l);
	printf("%d\n",t[0]);
	for(int i=1;i<=t[0];i++) printf("%d%c",t[i],i==t[0]?10:32);
}