思维 排序 + 线性筛

题意:有一个长度为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]] -- ;	
			}
		}
	}
}
发布了187 篇原创文章 · 获赞 12 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/Irving0323/article/details/94579785