题目大意:给出一个图,有n的顶点,m条边,所有的边都被涂成白色或者是黑色,白色为1,黑色为0,问我们可以找出一个生成树,是的这个生成树的边其中的白色数目为斐波那契数。
思路分析:
1.在看到这道题时,首先要明确一点的是,给出的边不一定能构成一个生成树,可以提前判断一次,在输入n和m之后,如果给出的m小于n-1,那么就可以直接输出no的,不用进行接下来的操作了。
2.这道题中,每个边的权值不是1就是0,那么如果可以建成一个生成树,那么路径和一定是有固定范围的,即 [ 0,n-1 ] ,而规定白边为1,那么我们可以以黑边优先求一次生成树(最小),再以白边求一次生成树(最大),得到的两个路径和,如果在这两个路径和的范围内存在斐波那契数,那么就可以输出yes了;那么为什么呢,我们求得的路径和一定代表了我们一共用了几条白边,在这个范围内的路径和,也就是不断用白边来代替黑边,所以,如果这个范围内如果有斐波那契数,那就代表可以构成题中要求的生成树。
代码实现:
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn=100010;
int n,m,fib[26],fa[maxn];
void F(){
fib[0]=1;
fib[1]=1;
for(int i=2;i<26;i++)
fib[i]=fib[i-1]+fib[i-2];
/*for(int i=1;i<26;i++)
printf("%d ",fib[i]);
printf("\n");*/
}
struct Node{
int f,t,val;
}edge[maxn];
int cmp1(Node a,Node b){
return a.val<b.val;
}
int cmp2(Node a,Node b){
return a.val>b.val;
}
int Find(int a){
if(a!=fa[a]) fa[a]=Find(fa[a]);
return fa[a];
}
void Merge(int a,int b){
int x=Find(a);
int y=Find(b);
fa[y]=x;
}
int Kruskal(){
for(int i=1;i<=n;i++)
fa[i]=i;
int sum=0,i,j;
for(i=0,j=0;i<n-1&&j<m;j++){
if(Find(edge[j].f)==Find(edge[j].t)) continue;
Merge(edge[j].f,edge[j].t);
sum+=edge[j].val;
i++;
}
if(i==n-1) return sum;
else return -1;
}
int main(){
F();
int T;
scanf("%d",&T);
int cnt=0;
while(T--){
cnt++;
scanf("%d%d",&n,&m);
for(int i=0;i<m;i++){
scanf("%d%d%d",&edge[i].f,&edge[i].t,&edge[i].val);
}
if(m<n-1){
printf("Case #%d: No\n",cnt);
continue;
}
sort(edge,edge+m,cmp1);
int ans1=Kruskal();
if(ans1==-1){
printf("Case #%d: No\n",cnt);
continue;
}
sort(edge,edge+m,cmp2);
int ans2=Kruskal();
int flag=0;
for(int i=ans1;i<=ans2;i++){
for(int j=1;j<26;j++){
if(i==fib[j]){
flag=1;
break;
}
if(fib[j]>i) break;
}
}
//printf("%d %d\n",ans1,ans2);
if(flag==1) printf("Case #%d: Yes\n",cnt);
else printf("Case #%d: No\n",cnt);
}
}
这么找斐波那契数时间会少些。。。
for(int i=1;i<26;i++){
if(fib[i]>=ans1&&fib[i]<=ans2){
flag=1;
break;
}
}