hdu 2682

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#define N  700
#define NN 2000010
int prime[NN];
int a[N];
struct node {
int x,y,w;
}ma[N*N];
void ispri() {
  int i,j;
  memset(prime,-1,sizeof(prime));
  prime[0]=1;
  prime[1]=1;
  for(i=2;i<=1500;i++)
  if(prime[i]==-1) {
    for(j=i*2;j<=2000000;j+=i)
        prime[j]=1;
  }
 /* int k=0;
  for(i=2;i<=1000000;i++)
    if(prime[i]==-1) {
    k++;
    printf("%d ",i);
    }
  printf("%d\n",k);*/
}
int pre[N];
int find(int x) {
if(x!=pre[x])
pre[x]=find(pre[x]);
return pre[x];
}
int cmp(const void *a,const void *b) {
return (*(struct node *)a).w-(*(struct node *)b).w;
}
int MIN(int a,int b) {
return a>b?b:a;
}
int main() {
    int n,i,j,k,t,ans,len,aa,bb;
    ispri();
    scanf("%d",&t);
    while(t--) {
        scanf("%d",&n);
        for(i=1;i<=n;i++)
            pre[i]=i;
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
            len=0;
            for(i=1;i<=n;i++)
                for(j=i+1;j<=n;j++)
                if(prime[a[i]]==-1||prime[a[j]]==-1||prime[a[i]+a[j]]==-1)
                    ma[len].x=i,ma[len].y=j,ma[len++].w=MIN(MIN(a[i],a[j]),fabs(a[i]-a[j]));
                    qsort(ma,len,sizeof(ma[0]),cmp);
                 /*   for(i=0;i<len;i++)
                        printf("%d ",ma[i].w);*/
                ans=0;k=0;
        for(i=0;i<len&&k<n-1;i++) {
            aa=find(ma[i].x);
            bb=find(ma[i].y);
            if(aa!=bb) {
                pre[aa]=bb;
                ans+=ma[i].w;
                k++;
            }
        }
        if(k==n-1)
            printf("%d\n",ans);
        else
            printf("-1\n");
     }


return 0;
}

转载于:https://www.cnblogs.com/thefirstfeeling/p/4410744.html