hdu 2682 Tree

题目大意：T组测试实例，每组实例有n个城市，每个城市有一个幸福指数V；如果 Va 或 Vb 或 Va+Vb 是素数的话，两个城市可以连接起来；将两个城市连接起来所需的花费为min(min(Va,Vb),|Va-Vb|)；如果所有的城市都能连接起来的话输出最小花费，否则输出“-1”

#include<iostream>
#include<vector>
#include<cmath>
#define maxn 605
#define pr 2000009
using namespace std;
int n;
const int inf=1<<31-1;
vector<int>mapp[maxn];
int vaule[maxn][maxn];
int visit[maxn];
int v[maxn];
int u[pr]={0};
void prime()
{
	int i,j;
	u[0]=1,u[1]=1;
	for(i=2;i<pr;i++)
	{
		if(!u[i])
		{
			for(j=2;i*j<pr;j++){u[i*j]=1;}
		}
	}
} 
void prim()//prim算法 
{
	int re=0,k=0;
	int d[maxn];
	fill(d,d+n+1,inf);
	fill(visit,visit+1+n,0);
	d[1]=0;
	while(1)
	{
		int v=-1;
		for(int i=1;i<=n;i++)
		{
			if(!visit[i]&&(v==-1||d[i]<d[v])) v=i;
		}
		if(v==-1||d[v]==inf) break;
		re+=d[v];
		visit[v]=1;
		k++;
		for(int i=0;i<mapp[v].size();i++)
		{
			int x=mapp[v][i];
			if(!visit[x]&&vaule[v][x]<d[x]) d[x]=vaule[v][x];
		}
	}
	if(k==n) cout<<re<<endl;
	else cout<<"-1"<<endl;
}
int main()
{
	cin.sync_with_stdio(false);
	prime();
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n;
		for(int i=0;i<=n;i++) mapp[i].clear();
		for(int i=1;i<=n;i++)
		{
			int x;
			cin>>v[i];
			for(int j=1;j<=i;j++)
			{
				if(!u[v[i]]||!u[v[j]]||!(u[v[i]+v[j]]))
				{
					mapp[i].push_back(j);
					mapp[j].push_back(i);
					int p=abs(v[i]-v[j]);
					vaule[i][j]=min(min(v[i],v[j]),p);
					vaule[j][i]=min(min(v[i],v[j]),p);
				}
			}
		}
		prim();
	}
	return 0;
}