HDU 4786 斐波那契生成树

  这道题的意思是给你一些边, 某些边被染成黑色, 还有一些边被染成白色, 现在问你没有一个生成树他的白边的数量是斐波那契数列中的数, 我们可以先选黑边得到最最小的白边数, 先选白边得到最大的白边数, 然后看看这个区间有没有斐波那契数即可, 注意没有生成树时的情况。。 代码如下:

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
int N, M;
struct edge { int u, v, c; }e[100000 + 100];
int ne;
bool whitefirst(const edge &a, const edge &b)
{
    return a.c > b.c;
}
bool blackfirst(const edge&a, const edge&b)
{
    return a.c < b.c;
}

int par[100000+100];
void init(int n)
{
    for(int i=0; i<=n; i++) par[i] = i;
}
int Find(int x)
{
    if(par[x] == x) return x;
    else return par[x] = Find(par[x]);
}

int kruskal()
{
    init(N);
    int m = 0;   //已经选了的边
    int num = 0;
    for(int i=0; i<ne; i++)
    {
        int u = e[i].u, v = e[i].v;
        u = Find(u), v = Find(v);
        if(u!=v)
        {
            par[u] = v;
            m++;
            //num += e[i].c;
            if(e[i].c == 1) num++;
        }
        if(m == N-1) break;
    }
    if(m != N-1) num = -1;
    return num;
}

int fib[100];

int main()
{
    int T;
    int kase = 0;
    scanf("%d", &T);
    fib[0] = 1;
    fib[1] = 2;
    for(int i=2; i<=30; i++)
        fib[i] = fib[i-1]+fib[i-2];
    while(T--)
    {
        scanf("%d%d", &N, &M);
        ne = 0;
        for(int i=0; i<M; i++)
        {
            int u, v , c;
            scanf("%d%d%d", &u, &v, &c);
            e[ne++] = (edge){u, v, c};
        }
        sort(e, e+ne, whitefirst);
        int num = kruskal();
        sort(e, e+ne, blackfirst);
        int num2 = kruskal();

        if(num==-1 || num2==-1)
        {
            printf("Case #%d: No
", ++kase);
            continue;
        }
        bool flog = false;
        for(int i=0; i<=30&&!flog; i++)
            if(fib[i]>=num2 && fib[i]<=num) flog=true;
        if(flog)
            printf("Case #%d: Yes
", ++kase);
        else
            printf("Case #%d: No
", ++kase);
    }
    return 0;
}
原文地址:https://www.cnblogs.com/xingxing1024/p/5254845.html