给出
N
≤
5
e
4
,
M
≤
1
e
5
N\leq5e4,M\leq1e5
N≤5e4,M≤1e5的无向连通图,一些边是白色,一些边是黑色的。然后求问满足恰好的
K
K
K条边是白色的最小生成树的值是多少。
是有一个比较新奇的思路,有博主把这个称为凸优化,具体做法是先给所有的白边都增加一个相同的权值,然后对这个新图求最小生成树。那么如果满足树上的白边恰好是
K
K
K条,那么就更新答案。
由于显然增加的权值越大,白边的数量就会越少。所以可以二分增加的权值。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=5e4+7;
const int M=1e5+7;
struct edge {
int u,v,w,col;
bool operator <(const edge &rhs) const {
if(w<rhs.w) return 1;
else if(w==rhs.w) return col<rhs.col;
else return 0;
}
}e[M];
int fa[N];
int n,m,nd;
int find(int x) {
if(fa[x]==x) return fa[x];
else return fa[x]=find(fa[x]);
}
pair<int,int> check(int x) {
for(int i=1;i<=m;i++) if(!e[i].col) e[i].w+=x;
sort(e+1,e+1+m);
for(int i=1;i<=n;i++) fa[i]=i;
int tot=0,white=0,ans=0;
for(int i=1;i<=m;i++) {
int U=find(e[i].u);
int V=find(e[i].v);
if(U!=V) {
fa[U]=V;
ans+=e[i].w;
if(!e[i].col) white++;
tot++;
}
if(tot==n-1) break;
}
for(int i=1;i<=m;i++) if(!e[i].col) e[i].w-=x;
return make_pair(white,ans);
}
int main() {
scanf("%d%d%d",&n,&m,&nd);
for(int i=1;i<=m;i++) {
int u,v,w,col;
scanf("%d%d%d%d",&u,&v,&w,&col);
u++,v++;
e[i].u=u;e[i].v=v;
e[i].w=w;e[i].col=col;
}
int l=-100,r=100,ans=-1;
while(l<=r) {
int mid=(l+r)>>1;
pair<int,int> result=check(mid);
if(result.first>=nd) l=mid+1,ans=result.second-result.first*mid;
else r=mid-1;
}
printf("%d\n",ans);
return 0;
}