HDU 4786 Fibonacci Tree 最小生成树 kruskal算法

2013年区域赛成都赛区的第二道签到题，题意即为给一个图，每条边可以涂黑色或者白色，求问图中去边以后最后的生成树中，白色边数量可不可以为Fibonacci数（fibonacci数为f1=1，f2=2，fn=fn-1+fn-2），可以输出Yes，不可以输出No，此题做法是最大生成树+最小生成树+二分，但是由于此题中点的数量为100000，所以prim算法无效，只能使用kruskal算法，先打表得到所有fibonacci数，然后求出此题的最大生成树和最小生成树，并且判定此图是否联通，如果不连通，直接输出No，之后二分寻找此区间内是否含有fibonacci数（PS：这个地方最大生成树最大也就到100000，其实fibonnacci数也就是30个，试了一下其实即使暴力枚举也是可以过得），含有输出Yes，不含输出No即可得到答案。。。具体代码如下：

#include<algorithm>
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int fib[30];
struct st
{
    int l,r,c;
}edge[100005];
int fa[100005];
int cmp(const void *a,const void *b)
{
    return (*(st*)b).c-(*(st*)a).c;
}
int getfa(int a)
{
    if(0==fa[a])
        return a;
    else
        return fa[a]=getfa(fa[a]);
}
int main()
{
//    freopen("in.txt","r",stdin);
    fib[0]=1,fib[1]=2;
    int flag;
    for(int i=2;fib[i-1]<=100000;i++)
    {
        fib[i]=fib[i-1]+fib[i-2];
    }
    int t,ti=1;
    scanf("%d",&t);
    while(t--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&edge[i].l,&edge[i].r,&edge[i].c);
        }
        memset(fa,0,sizeof(fa));
        qsort(edge+1,m,sizeof(edge[0]),cmp);
        int hi=0,sum=0;
        for(int i=1;i<=m;i++)
        {
            int x=getfa(edge[i].l);
            int y=getfa(edge[i].r);
            if(x!=y)
            {
                fa[x]=y;
                hi+=edge[i].c;
                sum++;
            }
        }
        if(sum!=n-1)
        {
            printf("Case #%d: No\n",ti);
            ti++;
            continue;
        }
        int lo=0;
        memset(fa,0,sizeof(fa));
        for(int i=m;i>0;i--)
        {
            int x=getfa(edge[i].l);
            int y=getfa(edge[i].r);
            if(x!=y)
            {
                fa[x]=y;
                lo+=edge[i].c;
            }
        }
        flag=0;
        int i=0;
        while(fib[i]<=hi)
        {
            if(fib[i]>=lo&&fib[i]<=hi)
            {
                flag=1;
                break;
            }
            i++;
        }
        if(flag==1)
            printf("Case #%d: Yes\n",ti);
        else
            printf("Case #%d: No\n",ti);
        ti++;
    }
    return 0;
}