Codeforces-959D Mahmoud and Ehab and another array construction task 素数筛+质因数分解

题意:

    给一个数组a，求大于或者等于数组a的字典序的数组b，其中b满足数组内任意数互质，bi>=2。并且是所有满足条件中的字典序最小的一个。

题解：

         这道构造题首先要考虑的就是如果a数组内的数互质的话，就不需要构造出b数组，直接输出a数组就可以了。那么我们怎么判断整个数组内的数是否互质呢？不能单单的左右判断就完事（意思就是假如，2,3,4判断是否互质，2和3不互质，3和4不互质，那么不互质，结果是错误的）。我们可以考虑对每个数质因数分解。如果出现有重复的就要从这个数开始构造。找到第一个比这个数大而且与前面的数互质的数，然后从小到大输出不重复的素数就可以了。这道题还有个坑点。。就是素数表的范围不要只打到1e5个，因为答案要的最大素数可能不在1e5个数内。另外需要的就是质因数分解的时候要先判断这个数的全部质因数是否有重复，再存放重复，不然的话就会出错，比如3个数据3，6，2。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;
const int MAXN=2e6+7;
int a[MAXN],b[MAXN];
bool prime[MAXN],c[MAXN];
int cnt;
void prim()
{
    int i, j;
    prime[0]=prime[1]=0,prime[2]=1;
    for(i=3; i<MAXN; i++){
        if(i%2) prime[i]=1;
        else prime[i]=0;//偶数都不为素数。 
    }

    for(i=3; i<=sqrt(MAXN); i++)
    {   if(prime[i]==1)
        for(j=i+i; j<MAXN; j+=i)//选出素数的倍数（因为素数的倍数能被整除，所以不是素数。） 
 	    prime[j]=0;
    }
    cnt=0;
//  for(i=2; i<1e7; i++)
//  if( prime[i]==1 )printf("%d %d\n",i,++cnt);
//  printf("\n");
	for(int i=2;i<MAXN;i++)
	{
		if(prime[i]==1)
		b[++cnt]=i;
//		if(cnt>1e5)//这个条件是错误的，因为最后要输出的最大的素数有可能不会出现在这1e5个素数内。因为这个点WA了一整天，靠。。。 
//		break;		
	}
}
bool check(int x)
{
	int temp=x;
	for(int i=2;i*i<=temp;i++)
	{
		if(temp%i==0)
		{
			if(c[i])
			return 0;
			while(temp%i==0)
			temp/=i;
		}
	}
	if(temp>1)
	{
		if(c[temp])
		return 0;
	}
	
	for(int i=2;i*i<=x;i++)
	{
		if(x%i==0)
		{
			c[i]=1;
			while(x%i==0)
			x/=i;
		}
	}	
	if(x>1)
	c[x]=1;
	return 1;
}
int main()
{
	int n;
	prim();
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	bool flag=false;
	int j;
	for(int i=1;i<=n;i++)
	{
		j=i;
		if(check(a[i]))
		printf("%d ",a[i]),c[a[i]]=1;
		else{
			flag=true;
			break;
		}
	}
	if(flag){
	
	for(int k=a[j];k<=1e7&&j<=n;k++)
	{
		if(check(k)){
			printf("%d ",k),c[k]=1,j++;
			break;	
		}
	}
	for(int k=1;k<=cnt&&j<=n;k++)
	{
		if(!c[b[k]]){
		printf("%d ",b[k]);
		j++;
		c[b[k]];	
		}	
	}
	printf("\n"); 
	}
}