hdu2682

Tree

Time Limit: 6000/2000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 2671    Accepted Submission(s): 828

Problem Description

There are N (2<=N<=600) cities,each has a value of happiness,we consider two cities A and B whose value of happiness are VA and VB,if VA is a prime number,or VB is a prime number or (VA+VB) is a prime number,then they can be connected.What's more,the cost to connecte two cities is Min(Min(VA , VB),|VA-VB|).
Now we want to connecte all the cities together,and make the cost minimal.

 

Input

The first will contain a integer t,followed by t cases.
Each case begin with a integer N,then N integer Vi(0<=Vi<=1000000).

 

Output

If the all cities can be connected together,output the minimal cost,otherwise output "-1";

 

Sample Input

2
5
1
2
3
4
5

4
4
4
4
4

 

Sample Output

4
-1

AC代码：
#include<stdio.h> 
#include<cmath>
#include<math.h>
#include<string.h>
#include<stdlib.h>
int vi[611];  
int prime[2000111];  
struct A  
{  
        int a,b;  
        int len;  
}eage[180000];  
int tot;  
//并查集  
int set[611];  
void build(int n)  
{  
        int i;  
        for(i=0;i<n;i++) 
                set[i]=i;  
}  
int find(int k)  
{  
        if(set[k]==k)  
                return k;  
        set[k]=find(set[k]);  
                return set[k];  
}  
void Union(int f1,int f2)  
{  
        set[f1]=f2;  
}  
int MIN(int a,int b)  
{  
        return a>b?b:a;  
}  
int cmp(const void *a,const void *b)  
{  
        struct A *c,*d;  
        c=(struct A *)a;  
        d=(struct A *)b;  
        return c->len-d->len;  
}  
int main()  
{  
        int T,n,i,l,temp,f1,f2,cost;
        memset(prime,-1,sizeof(prime));  
        prime[0]=prime[1]=0;  
        for(i=2;i<=2000000;i++)  
        {  
                if(prime[i]==-1)  
                {  
                        for(temp=2*i;temp<=2000000;temp+=i)  
                                prime[temp]=0;  
                }  
        }  
        scanf("%d",&T);  
        while(T--)  
        {  
                scanf("%d",&n);  
                for(i=0;i<n;i++) 
                        scanf("%d",&vi[i]);  
                tot=0;  
                for(i=0;i<n;i++)  
                for(l=i+1;l<n;l++)  
                {  
                        if(prime[vi[i]] || prime[vi[l]] || prime[vi[i]+vi[l]])  
                        {  
                                eage[tot].a=i;  
                                eage[tot].b=l;  
                                eage[tot].len=MIN(vi[i],MIN(vi[l],abs(vi[i]-vi[l])));  
                                tot++;  
                        }  
                }  
                qsort(eage,tot,sizeof(eage[0]),cmp);  
                cost=0;  
                build(n);  
                for(i=0;i<tot;i++)  
                {  
                        f1=find(eage[i].a);  
                        f2=find(eage[i].b);  
                        if(f1==f2)  
                                continue;  
                        cost+=eage[i].len;  
                        Union(f1,f2);  
                }  
                temp=0;  
                for(i=0;i<n;i++) 
                if(set[i]==i)  
                        temp++;  
                if(temp>1)  
                        printf("-1\n");  
                else     
                        printf("%d\n",cost);  
        }  
        return 0;  
}