组队赛第四场:最小生成树变形

来源:互联网 发布:java常见的设计模式 编辑:程序博客网 时间:2024/06/07 02:44

组队赛第四场:

HDU 4786 Fibonacci Tree 

思路:

这题比赛的时候想了好久,最后队友机智的想到了。

不过那时不是我敲的,现在敲的1A。

想好就容易了。

直接把1或者0当做边的权值,然后按边从小到大排序,然后算最小生成用到了几条白边,然后再按边从大到小排序,然后再算白边用了几条。然后最小和最大需要用到的白边都算出来了。如果在这最小最大区间中存在那个啥数列的话就是Yes,否则就是No。

为什么在这区间里面就是对的呢?刚开始我也想了好久,然后发现,因为白边权值是1,然后黑边是0,然后假设用到白边最小的是6,最大的是10,那么,我们可以用黑边去替代两条白边,生成树就是有8条白边了,而正好是我们要找的数列中的数。其实还是有点抽象……自己脑子想想吧……

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<map>#include<queue>#include<set>#include<cmath>#include<bitset>#define mem(a,b) memset(a,b,sizeof(a))#define lson i<<1,l,mid#define rson i<<1|1,mid+1,r#define INF 510010#define maxn 1000010using namespace std;typedef long long ll;typedef unsigned long long ull;int f[maxn],fib[92];struct abc{    int u,v,w;}a[maxn];bool cmp1(abc a,abc b){    return a.w<b.w;}bool cmp2(abc a,abc b){    return a.w>b.w;}void Fib(){    fib[1]=1;fib[2]=2;    for(int i=3;i<41;i++)        fib[i]=fib[i-1]+fib[i-2];}int find(int x){    return x==f[x]?x:f[x]=find(f[x]);}void init(){    for(int i=0;i<maxn;i++)        f[i]=i;}int kruscal(int n,int m){    init();    int i,sum=0,num=0,flag=1;    for(i=0;i<n&&flag;i++)    {        int x=find(a[i].u);        int y=find(a[i].v);        if(x!=y)        {            f[y]=x;            sum+=a[i].w;            num++;        }        if(num==m-1) flag=0;    }    if(!flag) return sum;    else return -1;}int main(){    //freopen("test.txt","r",stdin);    int t;    cin>>t;    Fib();    for(int kk=1;kk<=t;kk++)    {        int n,m,i;        scanf("%d%d",&n,&m);        for(i=0;i<m;i++)            scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].w);        sort(a,a+m,cmp1);        int sum1=kruscal(m,n);        sort(a,a+m,cmp2);        int sum2=kruscal(m,n);        int cnt1=lower_bound(fib+1,fib+41,sum1)-fib;        int cnt2=lower_bound(fib+1,fib+41,sum2)-fib;        if(sum1==-1) printf("Case #%d: No\n",kk);//不存在最小生成树        else if(cnt1!=cnt2||cnt1==cnt2&&sum1==sum2&&fib[cnt1]==sum1)            printf("Case #%d: Yes\n",kk);        else printf("Case #%d: No\n",kk);    }    return 0;}


0 0