bzoj3624 [Apio2008]免费道路（图论+并查集）

我们首先把必须加入的0边加进去（把所有1边全加进去，然后此时还能加进去的0边就是必须加的0边）
如果必须加的0边个数>K则无解。否则继续加0边直到K条。
如果加不到K条也无解。然后再加1边，能加就加进去即可。
不连通也是无解。
解决。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 20010
#define M 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,K,u[M],v[M],op[M],fa[N],tot;
bool mk[M];
inline int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
inline bool merge(int x,int y){
    int xx=find(x),yy=find(y);
    if(xx==yy) return 0;fa[xx]=yy;--tot;return 1;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();K=read();
    for(int i=1;i<=m;++i) u[i]=read(),v[i]=read(),op[i]=read();
    for(int i=1;i<=n;++i) fa[i]=i;tot=n;
    for(int i=1;i<=m;++i) if(op[i]) merge(u[i],v[i]);
    for(int i=1;i<=m;++i){
        if(op[i]) continue;bool f=merge(u[i],v[i]);
        if(f) mk[i]=1,--K;
    }if(K<0||tot!=1){puts("no solution");return 0;}
    for(int i=1;i<=n;++i) fa[i]=i;
    for(int i=1;i<=m;++i) if(mk[i]) merge(u[i],v[i]);
    for(int i=1;i<=m&&K;++i){
        if(op[i]||mk[i]) continue;bool f=merge(u[i],v[i]);
        if(f) mk[i]=1,--K;
    }if(K){puts("no solution");return 0;}
    for(int i=1;i<=m;++i){
        if(!op[i]) continue;bool f=merge(u[i],v[i]);
        if(f) mk[i]=1;
    }for(int i=1;i<=m;++i)
        if(mk[i]) printf("%d %d %d\n",u[i],v[i],op[i]);
    return 0;
}