题意:有一个长度为n的序列a,ai未知,已知长度为2*n的序列b。序列a经过下列操作后变成序列b。
(1)先令序列a和b相等。
(2)若ai是素数,则将第ai个素数添加到序列b中。例如ai=2,则将第2个素数3添加到序列b中。
(3)若ai不是素数,则将ai的最大约数添加到序列b中。
输出任意一种序列a,数据保证有解。n <= 2e5,bi <= 2750131。
题解:首先想到bi的最大约数 < bi < 第bi个素数。则将序列b从小到大排序。然后从后向前遍历序列bi。若bi是素数,则将bi和k从序列b中删除(第k个素数是bi)。若bi不是素数,则将bi和bi的最大约数从序列b中删除。
线性筛可以得到第i个素数,也能得到最大约数(通过与最小约数相除得到,代码中有体现)。
AC代码:
#include<bits/stdc++.h>
#define N 400005
#define M 3000005
typedef long long ll ;
using namespace std ;
int n ;
int a[N] ;
int num[M] ;
int max1 = 0 ;
bool vis[M] ;
int prime[M] , tot ;
int fprime[M] ;
int maxfrime[M] ;
void init()
{
int i , j ;
tot=0;
memset(vis,0,sizeof(vis));
for(i=2;i<M;i++)
{
//如果未标记则得到一个素数
if(!vis[i])
{
prime[++tot]=i;
fprime[i] = tot ;
}
//标记目前得到的素数的i倍为非素数
for(j = 1 ; j <= tot && prime[j] * i < M ; j ++)
{
if(vis[i*prime[j]] == 0)
maxfrime[i*prime[j]] = max(i , prime[j]) ; //得到最大约数
vis[i*prime[j]]=1;
if(i%prime[j]==0) //精髓 prime数组 中的素数是递增的,当 i 能整除 prime[j],那么 i*prime[j+1] 这个合数肯定被 prime[j] 乘以某个数筛掉
{
break;
}
}
}
}
int main()
{
int i , j ;
scanf("%d" , &n) ;
init() ;
memset(num , 0 , sizeof(num)) ;
for(i = 1 ; i <= 2 * n ; i ++)
{
scanf("%d" , &a[i]) ;
num[a[i]] ++ ;
max1 = max(max1 , a[i]) ;
}
sort(a + 1 , a + n + 1) ;
for(i = max1 ; i >= 1 ; i --)
{
while(num[i] > 0)
{
if(!vis[i])
{
printf("%d " , fprime[i]) ;
num[i] -- ;
num[fprime[i]] -- ;
}
else
{
printf("%d " , i) ;
num[i] -- ;
num[maxfrime[i]] -- ;
}
}
}
}