首页 > 代码库 > HDOJ--4786--Fibonacci Tree【生成树】
HDOJ--4786--Fibonacci Tree【生成树】
题意:给出n个点,m条边,和边的信息。边有两种颜色,白色和黑色,现要求构造一个生成树,看能否满足白边的数量是斐波那契数。
这道题比赛的时候,小白想到了一种方法:按边颜色排序后,先用白边优先建树,求出最大白边最大个数maxm,再用黑边优先建树,求出白边最小个数minm,看这两个范围内是否存在斐波那契数。
听上去感觉还挺有道理,但是不知道怎么证明正确性,后来想想,生成树构造完之后,再添加任意一条边都会产生回路,而产生回路之后就有边会被替换,而minm是最少的白边数,也就是minm个白边是不会被换掉的,maxm同理,所以中间的回路替换掉边总能保证用白边替换黑边,或者黑边替换白边,所以可以这么做。
我信心满满的开始敲,然后WA了。。因为我写完cmp函数后忘记写sort了,而且还过了样例。改过之后交一发,还是WA,后来发现数组开了110。。。而且提示是WA不是RE
另外,这道题如果不进行路径压缩,会超时
#include<cstring> #include<string> #include<fstream> #include<iostream> #include<iomanip> #include<cstdio> #include<cctype> #include<algorithm> #include<queue> #include<map> #include<set> #include<vector> #include<stack> #include<ctime> #include<cstdlib> #include<functional> #include<cmath> using namespace std; #define PI acos(-1.0) #define MAXN 110000 #define eps 1e-7 #define INF 0x7FFFFFFF #define seed 131 #define ll long long #define ull unsigned ll #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 struct node{ int u,v,col; }edge[MAXN]; int father[MAXN],fi[90]; int n,m,flag; bool cmp(node x,node y){ return x.col>y.col; } int find(int x){ int t = x; while(father[t]!=t){ t = father[t]; } int k = x; while(k!=t){ int temp = father[k]; father[k] = t; k = temp; } return t; } void solve(){ int i,j=0; int flag2 = 0; int minm,maxm; minm = maxm = 0; sort(edge,edge+m,cmp); for(i=0;i<m;i++){ int a = find(edge[i].u); int b = find(edge[i].v); if(a!=b){ if(edge[i].col==1) maxm++; father[a] = b; j++; if(j>=n-1){ flag2 = 1; break; } } } if(flag2==0){ return ; } for(i=1;i<=n;i++){ father[i] = i; } j = 0; for(i=m-1;i>=0;i--){ int a = find(edge[i].u); int b = find(edge[i].v); if(a!=b){ if(edge[i].col==1) minm++; father[a] = b; j++; if(j>=n-1) break; } } //cout<<minm<<" "<<maxm<<endl; for(i=1;i<45;i++){ if(fi[i]>=minm&&fi[i]<=maxm) { flag = 1; break; } } } int main(){ int i,j,k=1,t; int a,b,c; fi[1] = 1; fi[2] = 2; for(i=3;i<45;i++){ fi[i] = fi[i-1] + fi[i-2]; //cout<<fi[i]<<" "<<i<<" "<<endl; } scanf("%d",&t); while(t--){ flag = 0; scanf("%d%d",&n,&m); for(i=1;i<=n;i++) father[i] = i; for(i=0;i<m;i++){ scanf("%d%d%d",&a,&b,&c); edge[i].u = a; edge[i].v = b; edge[i].col = c; } solve(); if(flag) printf("Case #%d: Yes\n",k++); else printf("Case #%d: No\n",k++); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。