LA 6540 Fibonacci Tree

来源：互联网发布：类似弹个车的软件编辑：程序博客网时间：2024/06/09 19:07

以前做过的题···重新做一遍之后怎么做怎么wa···后来GG了···第二天看不知道为啥A了···难道我失忆了？

题意：无向图，边有黑色和白色两种颜色，求是否存在一个生成树中白边的个数是斐波那契数。

解法：并查集。对边按颜色进行排序，白边在前用并查集计算生成树中白边个数，再倒着算一遍，得到生成树的白边的最大值和最小值，判断其中有没有斐波那契数，注意要判断是否能构成生成树。

代码：

#include<stdio.h>#include<iostream>#include<algorithm>#include<string>#include<string.h>#include<math.h>#include<map>#include<queue>#include<set>#include<stack>#include<vector>#define LL long longusing namespace std;int father[100005], f[100010] = {0};struct node{    int u, v, c;} edge[100005];bool cmp(node a, node b){    return a.c > b.c;}int FIND(int a){    if(father[a] != a)        father[a] = FIND(father[a]);    return father[a];}int main(){    f[1] = 1;    int f1 = 1, f2 = 1, f3 = 2;    while(f3 < 100010)    {        f[f3] = 1;        f1 = f2;        f2 = f3;        f3 = f1 + f2;    }//斐波那契数打表    int T;    while(~scanf("%d", &T))    {        int cse = 1;        while(T--)        {            int n, m, ans = 1;            scanf("%d%d", &n, &m);            for(int i = 0; i < m; i++)                scanf("%d%d%d", &edge[i].u, &edge[i].v, &edge[i].c);            sort(edge, edge + m, cmp);            int l = 0, r = 0;            for(int i = 0; i <= n; i++)                father[i] = i;            for(int i = 0; i < m; i++)            {                int a, b;                a = FIND(edge[i].u);                b = FIND(edge[i].v);                if(a != b)                {                    r += edge[i].c;                    father[a] = b;                }            }//计算白边最大值r            int flag = FIND(1);            for(int i = 2; i <= n; i++)                if(FIND(i) != flag)                {                    ans = 0;                    break;                }//判断是否有生成树            if(ans)            {                ans = 0;                for(int i = 0; i <= n; i++)                    father[i] = i;                for(int i = m-1; i >= 0; i--)                {                    int a, b;                    a = FIND(edge[i].u);                    b = FIND(edge[i].v);                    if(a != b)                    {                        l += edge[i].c;                        father[a] = b;                    }                }//计算白边最小值l                for(int i = l; i <= r; i++)                    if(f[i])                    {                        ans = 1;                        break;                    }            }            if(ans)                cout << "Case #" << cse++ << ": Yes" << endl;            else                cout << "Case #" << cse++ << ": No" << endl;        }    }    return 0;}

最近想改变一下代码风格···结果连字都快不会打了orz

0 0