原题链接:https://codeforces.com/contest/1176/problem/D
本文链接:https://www.cnblogs.com/blowhail/p/11146761.html
题目大意是 两个个数列 a , b 相同 ,如果 ai 是素数,那么b数列里添加上第ai个素数(2为第一个),如果不是素数,那么b数列里添加上ai的最大因子。现在给出添加完之后的b数列,求出a数列。
大致思路: 先从大到小排序,然后判断是不是素数,如果是素数,那就遍历一遍,找到它是第几个素数,然后输出并记录下来。 如果不是素数,就直接输出,再求出最大的因子记录下来。
写超时了好多次,修改了很多地方,可读性估计不太好
代码如下
#include <cstdio> #include <iostream> #include <algorithm> #include <cstring> #include <string> #include <cmath> #define ll long long #define pi 3.1415926 using namespace std; int primes[200006],a[400005],b[3000005],p2[3000005];
//这里primes数组按顺序记录素数,p2是 素数为1 非素数为0
//a数组是存下输入的数列,b数组是记录每个数的个数 bool cmp (int x,int y) { return x>y; } void prime (int n) { int i,t, k,j; primes[1]=2; p2[2]=1; int f,p=2; for (i=3;i<=n;++i){ k=sqrt(i); f=1; for (t=2;t<=k;++t) if(i%t==0) { f=0; break; } if (f){ primes[p++]=i; //按顺序存 p2[i]=1; // } } } int main () { int i,t,m,n,l,k,j,p; scanf("%d",&n); j=2*n; for(i=0;i<j;++i){ scanf("%d",&a[i]); b[a[i]]++; //记录每个数个数 } sort(a,a+j,cmp); prime(a[0]); int c=200000; //这里我用c来记录每次遍历素数的位置,然后下一次就直接从这个位置继续遍历(因为已经从大到小排序了) for (i=0;i<j;++i) { if (b[a[i]]==0) continue; if (p2[a[i]]==1){ for (int h=c;h>=1;--h) if (primes[h]==a[i]){ c=h; break; }
printf("%d ",c);
b[ a[i] ]--; b[c]--; //b是记录每个数出现次数,每次遍历完都把找出来的数给减去 } else { printf("%d ",a[i]); for (int h=2;h<a[i];++h) if (a[i]%h==0) { k=a[i]/h; break; } b[a[i]]--; b[k]--; } } printf("\n"); return 0; }