bzoj3332: 旧试题

这题就是最大生成树。

把两个点之间的期望建边排序。

把相同的期望一起做，那么在这个做之前，这些有着相同期望的点两两肯定不连，否则就输出No了。

相同的做完之后，再次for一遍check一下有没有两两之间还是不能连的，有那么输出No。

其他的就是直接并查集。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;

int n,m;
bool mp[][];
struct node
{
    int x,y,d;
}a[];int len;
void ins(int x,int y,int d)
{
    len++;
    a[len].x=x;a[len].y=y;a[len].d=d;
}
bool cmp(node n1,node n2)
{
    if(n1.d>n2.d)return true;
    return false;
}
void init()
{
    scanf("%d%d",&n,&m);

    int x,y;
    memset(mp,false,sizeof(mp));
    for(int i=;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        mp[x][y]=true;mp[y][x]=true;
    }

    len=;
    for(int i=;i<=n;i++)
        for(int j=;j<=n;j++)
        {
            scanf("%d",&x);
            if(i<j)
                ins(i,j,x);
        }
    sort(a+,a+len+,cmp);
}

//-------------init---------------

int fa[];
int findfa(int x)
{
    if(fa[x]==x)return x;
    fa[x]=findfa(fa[x]);return fa[x];
}

int top,stax[],stay[];
int main()
{
    int T;
    scanf("%d",&T);
    for(int tt=;tt<=T;tt++)
    {
        init();
        for(int i=;i<=n;i++)fa[i]=i;

        bool bk=true;top=;int last=;
        for(int i=;i<=len;i++)
        {
            if(last!=a[i].d)
            {
                last=a[i].d;

                for(int j=;j<=top;j++)
                    if(findfa(stax[j])!=findfa(stay[j]))
                        {bk=false;break;}
                top=;
                if(bk==false)break;

                for(int j=i;a[j].d==a[i].d&&j<=len;j++)
                    if(findfa(a[j].x)==findfa(a[j].y))
                        {bk=false;break;}
                if(bk==false)break;
                if(a[i].d==-)break;
            }

            int x=a[i].x,y=a[i].y;
            int fx=findfa(x),fy=findfa(y);
            if(fx!=fy)
            {
                if(mp[x][y]==true)fa[fx]=fy;
                else
                {
                    top++;
                    stax[top]=a[i].x;
                    stay[top]=a[i].y;
                }
            }
        }
        if(bk==true)printf("Case #%d: Yes\n",tt);
        else         printf("Case #%d: No\n",tt);
    }
    return ;
}