HDU4786(生成树)

题目大意：给你N个点，M条边，权值为1的是白边，权值为0的边是黑边。问能否找到一棵生成树，使生成树的白边个数恰好为斐波那契数。

思路：求一遍最大生成树，在求一遍最小生成树，如果在两个值之间有斐波那契数的话，输出Yes，否则输出No。当然，如果本身就不连通的话，就输出No。

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
#include <iostream>
using namespace std;
const int maxn = 111111;
int fa[maxn];
int n, m;
int fibo[55];
struct node
{
    int from, to;
    int w;
}p[maxn];
void fib()
{
    fibo[0] = 1;
    fibo[1] = 2;
    for(int i=2; i<=50; i++)
    {
        fibo[i] = fibo[i-1] + fibo[i-2];
    }
}
bool cmp1(node a, node b)
{
    return a.w < b.w;
}
bool cmp2(node a, node b)
{
    return a.w > b.w;
}
void init()
{
    for(int i=0; i<=n; i++)
    {
        fa[i] = i;
    }
}
int find_fa(int x)
{
    return fa[x] == x ? fa[x] : fa[x] = find_fa(fa[x]);
}
int kruskal()
{
    int ans = 0;
    int cnt = 0;
    for(int i=0; i<m; i++)
    {
        int fx = find_fa(p[i].from);
        int fy = find_fa(p[i].to);
        if(fx != fy)
        {
            fa[fx] = fy;
            ans += p[i].w;
            cnt++;
        }
        if(cnt == n-1)
            break;
    }
    if(cnt == n-1)
        return ans;
    else
        return 0;//判断是否连通
}
int main()
{
    int T, kase = 0;
    fib();
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d", &n, &m);
        init();
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d", &p[i].from, &p[i].to, &p[i].w);
        }
        sort(p, p+m, cmp1);
        int min_mst = kruskal();
        init();
        sort(p, p+m, cmp2);
        int max_mst = kruskal();
        int flag = 0;
        for(int i=min_mst; i<=max_mst; i++)
        {
            for(int j=0; j<=50; j++)
            {
                if(i == fibo[j])
                {
                    flag =1;
                    break;
                }
            }
        }
        printf("Case #%d: ", ++kase);
        if(flag)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}