您的位置:首页 > 其它

HDU 4786 斐波那契生成树

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

#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\n", ++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\n", ++kase);
else
printf("Case #%d: No\n", ++kase);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: