hdu 2682(最小生成树）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2682

思路：一个裸的最小生成树，先打个素数表，然后就是连边了。dfs判断是否连通，连通的话就是kruskal求最小生成树了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define MAXN 666
struct Edge{
   int u,v,w;
}edge[MAXN*MAXN];
bool map[MAXN][MAXN];
bool prime[2222222];
int parent[MAXN];
bool mark[MAXN];
int value[MAXN];
int n,m,sum;

void Initiate(){
   memset(prime,true,sizeof(prime));
   prime[0]=prime[1]=false;
   for(int i=2;i*i<=2222000;i++){
      if(prime[i]){
         for(int j=i*i;j<=2222000;j+=i){
            prime[j]=false;
         }
      }
   }
}

void dfs(int u){
   mark[u]=true;
   for(int i=1;i<=n;i++){
      if(map[u][i]&&!mark[i])dfs(i);
   }
}

int cmp(const Edge &p,const Edge &q){
   return p.w<q.w;
}

int Find(int x){
   int s;
   for(s=x;parent[s]>=0;s=parent[s])
   ;
   while(s!=x){
      int tmp=parent[x];
      parent[x]=s;
      x=tmp;
   }
   return s;
}

void Union(int R1,int R2){
   int r1=Find(R1);
   int r2=Find(R2);
   if(r1==r2)return ;
   if(parent[r1]<parent[r2]){
      parent[r1]+=parent[r2];
      parent[r2]=r1;
   }else {
      parent[r2]+=parent[r1];
      parent[r1]=r2;
   }
}


int main(){
 //  freopen("1.txt","r",stdin);
   Initiate();
   int _case;
   scanf("%d",&_case);
   while(_case--){
      scanf("%d",&n);sum=m=0;
      for(int i=1;i<=n;i++)scanf("%d",&value[i]);
      memset(mark,false,sizeof(mark));
      memset(map,false,sizeof(map));
      for(int i=1;i<=n;i++){
         for(int j=i+1;j<=n;j++){
            if(prime[value[j]]||prime[value[i]]||prime[value[i]+value[j]]){
               map[i][j]=map[j][i]=true;
               edge[m].u=i;
               edge[m].v=j;
               edge[m++].w=min(min(value[i],value[j]),abs(value[i]-value[j]));
            }
         }
      }
      dfs(1);
      bool flag=true;
      for(int i=1;i<=n;i++)if(!mark[i]){ flag=false;break; }
      if(!flag){ puts("-1");continue; }
      memset(parent,-1,sizeof(parent));
      sort(edge,edge+m,cmp);
      for(int i=0;i<m;i++){
         int u=Find(edge[i].u);
         int v=Find(edge[i].v);
         if(u!=v){
            sum+=edge[i].w;
            Union(u,v);
         }
      }
      printf("%d\n",sum);
   }
   return 0;
}