HDU 4786 Fibonacci Tree(生成树问题)

分析：其实就是求最大生成树和最小生成树、然后看之间有没有斐波那契数、

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
#define INF 0x7ffffff
const int N=100005;
int m,n;
int temp;
struct node
{
    int u,v,val;
}edge[N];
int f[N];
int father[N];
int cmp(node a,node b)
{
    if(a.val<b.val)return 1;
    else return 0;
}
void init()
{
    for(int i=1;i<=n;i++)
        father[i]=i;
}
int find(int x)
{
    int r=x;
    while(r!=father[r])
        r=father[r];
    int b=x;
    int f;
    while(b!=r)
    {
        f=father[b];
        father[b]=r;
        b=f;
    }
    return r;
}
int merge(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    {
        father[fx]=fy;
        temp--;
        return 1;
    }
    return 0;
}
int main()
{
    f[1]=1;
    f[2]=2;
    for(int i=3;i<10000;i++)
    {
        f[i]=f[i-1]+f[i-2];
    }
    int t;
    int cas=1;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].val);
        }
        sort(edge,edge+m,cmp);
       // printf("%d\n",edge[0].val);
        int minx=0;
        temp=n-1;
        //printf("temp  %d \n",temp);
        for(int i=0;i<m;i++)
        {
            int ans=merge(edge[i].u,edge[i].v);
            if(ans)minx+=edge[i].val;
        }
        if(temp>0)
        {
            //printf("cao\n");
            printf("Case #%d: ",cas++);
            printf("No\n");
            continue;
        }
        int maxx=0;
        init();
        for(int i=m-1;i>=0;i--)
        {
            int ans=merge(edge[i].u,edge[i].v);
            if(ans)maxx+=edge[i].val;
        }
        //printf("%d %d\n",minx,maxx);
        int flag=0;
        for(int i=1;i<10000;i++)
        {
            if(f[i]>=minx&&f[i]<=maxx)
            {
                flag=1;break;
            }
        }
        printf("Case #%d: ",cas++);
        if(flag)printf("Yes\n");
        else printf("No\n");
    }

    return 0;
}