hdu2682

这个事并查集。

刚开始以为会超时，但，，，，1A了。

就是把能连的都枚举出来，然后构建最小生成树即可；

 

 

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
typedef struct 
{
    int x,y,s;
}A;
A a[360003];
int su[2000001]={0},d=0;
void sushu()  //产生2-n素数序列，结果在su[]里；
{
    int i,k;
    su[0]=su[1]=1;
    for(i=2;i<2000001;i++)
        if(su[i]==0)
        {
            for(k=i+i;k<=2000001;k+=i)
                su[k]=1;
        }
        
}
int min1 (int a,int b)
{
    return a>b?b:a;
}
int abs1(int a)
{
    return a>0? a:-a;
}
int cmp(const void *p1,const void *p2)
{
    A *q1=(A *)p1;
    A *q2=(A *)p2;
    return q1->s-q2->s;
}
__int64 res[603];
int father[603];      //并查集需要预处理father[n]=n;
int find(int n)  //找到头结点
{
    if(n!=father[n])
        father[n]=find(father[n]);
    return father[n];
}
int un (int i) //合并；
{
    int x,y;
    x=find(a[i].x);
    y=find(a[i].y);
    if(x!=y)
    {
        if(x<y)
        {
            father[y]=x;
            res[x]+=a[i].s+res[y];
        }
        else
        {
            father[x]=y;
            res[y]+=a[i].s+res[x];
        }
        return 1;
    }
    return 0;
}

int main ()
{
    
    int t,n,i,k,x[603],m,f;
    __int64 min;
    sushu();
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        m=0;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&x[i]);
            if(!su[x[i]]) //枚举所以边。
            {
                for(k=1;k<i;k++)
                {
                    a[m].x=i;
                    a[m].y=k;
                    a[m].s=min1(abs1(x[i]-x[k]),min1(x[i],x[k]));
                    m++;
                }
            }
            else
            {
                for(k=1;k<i;k++)
                    if(su[x[k]]==0||su[x[k]+x[i]]==0)
                {
                    a[m].x=i;
                    a[m].y=k;
                    a[m].s=min1(abs1(x[i]-x[k]),min1(x[i],x[k]));
                    m++;
                }
            }
        }
        qsort(a,m,sizeof(a[0]),cmp);//按边长排序；
        for(i=0;i<=n;i++)
        {
            father[i]=i;
            res[i]=0;
        }
        for(i=0;i<m;i++)//并查集找最小生成树；
            un(i);
        min=-1;
        for(i=1,f=0;i<n;i++) // 如果能找到最小生成树，输出结果，不能输出-1；
            if(father[i]==i)
            {
                f++;
                if(min==-1||res[i]<min)
                    min=res[i];
            }
        if(f==1)
            printf("%I64d\n",min);
        else
            puts("-1");

    }
    return 0;
}