Count Prime Pairs
单点时限: 2.0 sec
内存限制: 512 MB
对于数组a,如果i≠j并且ai+aj是一个质数,那么我们就称(i,j)为质数对,计算数组中质数对的个数。
输入格式
第一行输入一个n,表示数组的长度,接下来n个整数,第i个数代表ai。
(1≤n≤100000,0≤ai≤100)
输出格式
输出数组中质数对的个数。
样例
input
3
1 2 3
output
4
提示
样例说明:a1+a2,a2+a1,a2+a3,a3+a2都为质数,总共有四对。
这道题n的范围很大,ai的范围很小,所以两个数的和最大才是200,可以先素数打表,再把输入的100以内的数的个数存一下,然后将素数数组中每个数和数列中的进行判断。
#include<iostream>
#include<algorithm>
#include<bits/stdc++.h>
using namespace std;
int prime[210];
bool check[210];
int num=0;
void isprime(){
memset(check,true,sizeof(check));
for(int i=2;i<=210;i++){
if(check[i]){
prime[++num]=i;
}
for(int j=1;(j<=num)&&(i*prime[j]<=210);j++){
check[i*prime[j]]=false;//cout<<"tuy";
//if(i%prime[j]==0) break;
}
}
//for(int i=1;i<num;i++){
// cout<<prime[i]<<" ";
//}
}
int main(){
long long int t,a[100010];
cin>>t;
int shu[210]={0};
for(int i=0;i<t;i++){
cin>>a[i];
shu[a[i]]++;
}
long long int sum=0;
isprime();
for(int i=1;i<=num;i++){
//cout<<prime[i]<<endl;
int now=prime[i];
for(int j=0;j<t;j++){
if(now>=a[j]){
if(now==(a[j]*2)){
// cout<<a[j]<<endl;
sum+=shu[a[j]]-1;
}
else{
if(shu[a[j]]&&shu[now-a[j]])
{
//cout<<a[j]<<" "<<now-a[j]<<" "<<now<<endl;
sum+=shu[now-a[j]];
}
}
}
}
}
cout<<sum<<endl;
//cout<<check[2]<<check[3]<<check[4];
/*for(int i=0;i<210;i++){
for(int j=i+1;j<210;j++){
if(shu[i]&&shu[j]){
if(check[i+j]){
sum+=(shu[j-i]);
if(j=2*i) sum-=1;
}
// cout<<a[i]<<" "<<a[j];
}
}
}
cout<<sum*2<<endl;*/
return 0;
}