題目大意:給你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;
}