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