题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2682
思路：一个裸的最小生成树，先打个素数表，然后就是连边了。dfs判断是否连通，连通的话就是kruskal求最小生成树了。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<cmath> 6 using namespace std; 7 #define MAXN 666 8 struct Edge{ 9 int u,v,w; 10 }edge[MAXN*MAXN]; 11 bool map[MAXN][MAXN]; 12 bool prime[2222222]; 13 int parent[MAXN]; 14 bool mark[MAXN]; 15 int value[MAXN]; 16 int n,m,sum; 17 18 void Initiate(){ 19 memset(prime,true,sizeof(prime)); 20 prime[0]=prime[1]=false; 21 for(int i=2;i*i<=2222000;i++){ 22 if(prime[i]){ 23 for(int j=i*i;j<=2222000;j+=i){ 24 prime[j]=false; 25 } 26 } 27 } 28 } 29 30 void dfs(int u){ 31 mark[u]=true; 32 for(int i=1;i<=n;i++){ 33 if(map[u][i]&&!mark[i])dfs(i); 34 } 35 } 36 37 int cmp(const Edge &p,const Edge &q){ 38 return p.w<q.w; 39 } 40 41 int Find(int x){ 42 int s; 43 for(s=x;parent[s]>=0;s=parent[s]) 44 ; 45 while(s!=x){ 46 int tmp=parent[x]; 47 parent[x]=s; 48 x=tmp; 49 } 50 return s; 51 } 52 53 void Union(int R1,int R2){ 54 int r1=Find(R1); 55 int r2=Find(R2); 56 if(r1==r2)return ; 57 if(parent[r1]<parent[r2]){ 58 parent[r1]+=parent[r2]; 59 parent[r2]=r1; 60 }else { 61 parent[r2]+=parent[r1]; 62 parent[r1]=r2; 63 } 64 } 65 66 67 int main(){ 68 // freopen("1.txt","r",stdin); 69 Initiate(); 70 int _case; 71 scanf("%d",&_case); 72 while(_case--){ 73 scanf("%d",&n);sum=m=0; 74 for(int i=1;i<=n;i++)scanf("%d",&value[i]); 75 memset(mark,false,sizeof(mark)); 76 memset(map,false,sizeof(map)); 77 for(int i=1;i<=n;i++){ 78 for(int j=i+1;j<=n;j++){ 79 if(prime[value[j]]||prime[value[i]]||prime[value[i]+value[j]]){ 80 map[i][j]=map[j][i]=true; 81 edge[m].u=i; 82 edge[m].v=j; 83 edge[m++].w=min(min(value[i],value[j]),abs(value[i]-value[j])); 84 } 85 } 86 } 87 dfs(1); 88 bool flag=true; 89 for(int i=1;i<=n;i++)if(!mark[i]){ flag=false;break; } 90 if(!flag){ puts("-1");continue; } 91 memset(parent,-1,sizeof(parent)); 92 sort(edge,edge+m,cmp); 93 for(int i=0;i<m;i++){ 94 int u=Find(edge[i].u); 95 int v=Find(edge[i].v); 96 if(u!=v){ 97 sum+=edge[i].w; 98 Union(u,v); 99 } 100 } 101 printf("%d\n",sum); 102 } 103 return 0; 104 }