hdu 2682(最小生成树）

题目链接：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 }

View Code