Codeforces 1176D Recover it!

傳送門

思路:每次考慮最大的數字即可。最大的數字如果是質數,那麼它肯定輸入後來附加到數組b的,那麼他在質數中的位置下標肯定就是數組a的元素。最大的數字如果不是質數,那麼他肯定在數組a,因爲它不可能是任何數的因子。

 

#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <string>
#include <queue>
#include <cmath>
#include <stack>
#include <time.h>
#include <map>
#include <set>
#define mem(a,x) memset(a,x,sizeof(a))
#define gi(x) scanf("%d",&x)
#define gi2(x,y) scanf("%d%d",&x,&y)
#define gi3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define gll(x) scanf("%lld",&x)
#define gll2(x,y) scanf("%lld%lld",&x,&y)
using namespace std;
const double eps=1e-8; 
typedef long long ll;
const int MAXN=2750135;
const ll mod=1e9+7;
const int inf=0x3f3f3f3f;
int n;
int a[400005];
int book[MAXN];
int prime[MAXN];
int loc[MAXN];
int maxn=0;
vector<int>ans;
vector<int>pri;
void init(){
	pri.push_back(-1);
	int num=1;
	for(int i=2;i<MAXN;i++){
		if(prime[i]==0){
			for(int j=i*2;j<MAXN;j+=i){
				if(prime[j]==0){
					prime[j]=i;
				}
			}
			loc[i]=num++;
			//cout<<i<<' '<<loc[i]<<endl;
		}
	}
}

int main(){
	
	gi(n);
	
	for(int i=1;i<=2*n;i++){
		gi(a[i]);
		maxn=max(maxn,a[i]);
		book[a[i]]++;
	}
	init();
	for(int i=maxn;i>=2;i--){
		int t=book[i];
		if(t>0){
			if(prime[i]==0){
				int l=loc[i];
			//	cout<<i<<' '<<l<<endl;
				book[l]-=t;
				for(int j=0;j<t;j++){
					ans.push_back(l);
				}
			}
			else{
				int l=i/prime[i];
				book[l]-=t;
				for(int j=0;j<t;j++){
					ans.push_back(i);
				}
			}
		}
	}
	for(int i=0;i<ans.size();i++){
		printf("%d ",ans[i]);
	}
	
	
	
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章