有没有第一个点的数据，老过不掉

#include <iostream>
#include <cstdio>
#include <cmath>
#include <queue>
#include <algorithm>
#define LL long long
#define INF (1<<30)
using namespace std;

struct MergeFindSet{
    int F[20200];
    void pre(int x){
    for(int i=1;i<=x;i++)
        F[i]=i;
    }
    int findf(int x){
        return x==F[x]?x:F[x]=findf(F[x]);
    }
    void mergef(int x,int y){
        F[findf(x)]=findf(y);
    }
}S,T;

int edge[100100][3],tot;
bool flag[100100];
int main(){
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    S.pre(n);T.pre(n);
    int n_k=0;
    for(int i=1;i<=m;i++){
        int a,b,c;scanf("%d%d%d",&a,&b,&c);
        if(c==1)S.mergef(a,b);
        edge[i][0]=a;
        edge[i][1]=b;
        edge[i][2]=c;
    }
    for(int i=1;i<=m;i++)if(edge[i][2]==0){
        int a=edge[i][0];
        int b=edge[i][1];
        if(S.findf(a)!=S.findf(b)&&n_k<k){
            S.mergef(a,b);
            T.mergef(a,b);
            n_k++;
            flag[i]=true;
        }
    }
    for(int i=2;i<=n;i++)
        if(S.findf(1)!=S.findf(i)){
            printf("no solution");
            return 0;
        }
    for(int i=1;i<=m;i++)if(edge[i][2]==0){
        if(n_k==k)break;
        if(T.findf(edge[i][0])!=T.findf(edge[i][1])){
            T.mergef(edge[i][0],edge[i][1]);
            n_k++;
            flag[i]=true;
        }
        if(n_k==k)break;
    }
    if(n_k<k){
        for(int i=1;i<=m;i++)
            if(edge[i][2]==0&&flag[i]==false){
                flag[i]=true;
                n_k++;
                if(n_k==k)break;
            }
    }
    for(int i=1;i<=m;i++)if(edge[i][2]==1){
        if(T.findf(edge[i][0])!=T.findf(edge[i][1])){
            T.mergef(edge[i][0],edge[i][1]);
            flag[i]=true;
        }
    }
    for(int i=2;i<=n;i++)
        if(T.findf(1)!=T.findf(i)){
            printf("no solution");
            return 0;
        }
    for(int i=1;i<=m;i++)if(flag[i])
        printf("%d %d %d\n",edge[i][0],edge[i][1],edge[i][2]);
    return 0;
}