hdu4786Fibonacci Tree

链接：http://acm.hdu.edu.cn/showproblem.php?pid=4786

题意：给定n个点m条边，边分为黑白两种边。问是否存在一棵生成树使得树中的白边数量是一个斐波那契数。

分析：很明显可行的斐波那契数很少才24个，但是怎么判断当前这个数是否能是一棵生成树中的白边数呢？首先肯定是要判断所有边是否能生成树。然后最少要用多少条白边呢？用完所有的黑边还缺多少条边生成树就差多少白边啦。最多能用多少条白边呢？用完所有白边能生成多少树上的边那就最多能用多少咯。然后我们只需要判断白边的范围内是否存在一个斐波那契数就行啦。

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=100010;
const int mod=100000000;
const int MOD1=1000000007;
const int MOD2=1000000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const int MAX=2000000010;
const ll INF=1ll<<55;
const double pi=acos(-1.0);
typedef double db;
typedef unsigned long long ull;
struct node {
    int x,y,w;
}a[N];
int n,f[N],h[30];
int cmd(node a,node b) {
    return a.w<b.w;
}
int find_f(int x) {
    return f[x]==x ? f[x]:f[x]=find_f(f[x]);
}
int get(int l,int r) {
    int i,fa,fb,ret=0;
    for (i=1;i<=n;i++) f[i]=i;
    for (i=l;i<=r;i++) {
        fa=find_f(a[i].x);fb=find_f(a[i].y);
        if (fa!=fb) {
            ret++;f[fb]=fa;
        }
    }
    return ret;
}
int main()
{
    int i,g,m,t,bo,ca,mi,mx;
    scanf("%d", &t);
    h[1]=1;h[2]=2;
    for (i=3;i<30;i++) h[i]=h[i-1]+h[i-2];
    for (ca=1;ca<=t;ca++) {
        scanf("%d%d", &n, &m);
        for (g=0,i=1;i<=m;i++) {
            scanf("%d%d%d", &a[i].x, &a[i].y, &a[i].w);
            if (a[i].w==0) g++;
        }
        sort(a+1,a+m+1,cmd);
        printf("Case #%d: ", ca);
        if (get(1,m)!=n-1) printf("No\n");
        else {
            mi=n-1-get(1,g);
            mx=get(g+1,m);bo=0;
            for (i=1;i<=24;i++)
            if (mi<=h[i]&&h[i]<=mx) { bo=1;break ; }
            if (bo) printf("Yes\n");
            else printf("No\n");
        }
    }
    return 0;
}