Tree
Time Limit : 6000/2000ms (Java/Other) Memory Limit : 32768/32768K (Java/Other)
Total Submission(s) : 26 Accepted Submission(s) : 10
Font: Times New Roman | Verdana | Georgia
Font Size: ← →
Problem Description
There are N (2<=N<=600) cities,each has a value of happiness,we consider two cities A and B whose value of happiness are VA and VB,if VA is a prime number,or VB is a prime number or (VA+VB) is a prime number,then they can be connected.What's more,the cost to
connecte two cities is Min(Min(VA , VB),|VA-VB|).
Now we want to connecte all the cities together,and make the cost minimal.
Now we want to connecte all the cities together,and make the cost minimal.
Input
The first will contain a integer t,followed by t cases.
Each case begin with a integer N,then N integer Vi(0<=Vi<=1000000).
Each case begin with a integer N,then N integer Vi(0<=Vi<=1000000).
Output
If the all cities can be connected together,output the minimal cost,otherwise output "-1";
Sample Input
2 5 1 2 3 4 5 4 4 4 4 4
Sample Output
4 -1
Author
最小生成树:类似于畅通工程一类的,不过中间多加了素数的处理判断
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
struct inin
{
int a;
int b;
int dis;
}boy[410000];
int p[660];
int prime[1000010];
int p2[660];
int cmp(inin a,inin b)
{
return a.dis<b.dis;
}
int find(int n)
{
return p[n]==n ? n : p[n] = find(p[n]);
}
int main()
{
int T;
int n;
int k;
int re;
int sum;
int i,j;
prime[0]=1;
prime[1]=1;
prime[2]=0;
for(i=2;i<1000000;i++)
{
if(!prime[i])
{
for(j=i*2;j<1000000;j+=i)
{
prime[j]=1;
}
}
}
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
memset(p2,0,sizeof(p2));
memset(boy,0,sizeof(boy));
for(i=0;i<=n;i++)
{
p[i]=i;
}
for(i=1;i<=n;i++)
{
scanf("%d",&p2[i]);
}
k=0;
for(i=1;i<=n;i++)
{
for(j=i+1;j<=n;j++)
{
int temp=p2[i]-p2[j];
if(temp<0) temp=-temp;
// if(i!=j)
if(!prime[p2[i]]||!prime[p2[j]]||!prime[p2[i]+p2[j]])
{
boy[k].a=i;
boy[k].b=j;
boy[k++].dis=min(p2[i],min(p2[j],temp));
}
}
}
sort(boy,boy+k,cmp);
sum=0; re=1;
for(i=0;i<k;i++)
{
int f1=find(boy[i].a);
int f2=find(boy[i].b);
if(f1!=f2)
{
sum+=boy[i].dis;
p[f1]=f2;
re++;
}
}
if(re!=n)
{
printf("-1\n");
}
else
{
printf("%d\n",sum);
}
}
return 0;
}