woj 1566 Spanning Tree (KRU)

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<stdlib.h>
#include<algorithm>
#include<queue>
#include<vector>
#include<ctype.h>
#define LL __int64
using namespace std;
const int MAXN=100000+5;
const int MAX=1000+5;
struct node
{
    int s;
    int e;
    int val;
    bool operator<(const node A) const
    {
        return val<A.val;
    }
}a[MAXN],b[MAX];
int p[MAX],ans,cnt;
int n,m,q;

int findfa(int x)
{
    p[x]==x?x:p[x]=findfa(p[x]);
}
int kru()
{
    int res=0;
    sort(a,a+m);
    for(int i=1;i<=n;i++) p[i]=i;

    for(int i=0;i<m;i++)
    {
        int x=findfa(a[i].s);
        int y=findfa(a[i].e);
        if(x!=y)
        {
            p[x]=y;
            b[cnt++]=a[i];
            res+=a[i].val;
        }
    }
    return res;
}

int kru1(node now)
{
    int res=0;
    cnt=0;
    b[n-1]=now;
    sort(b,b+n);
    for(int i=1;i<=n;i++) p[i]=i;

    for(int i=0;i<n;i++)
    {
        int x=findfa(b[i].s);
        int y=findfa(b[i].e);
        if(x!=y)
        {
            p[x]=y;
            b[cnt++]=b[i];
            res+=b[i].val;
        }
    }
    return res;
}
int main()
{
    while(scanf("%d %d %d",&n,&m,&q)!=EOF)
    {
        for(int i=0;i<m;i++)
            scanf("%d %d %d",&a[i].s,&a[i].e,&a[i].val);

        cnt=0;
        kru();
        while(q--)
        {
            node now;
            scanf("%d %d %d",&now.s,&now.e,&now.val);
            ans=kru1(now);
            printf("%d
",ans);
        }
    }
    return 0;
}