Problem - 1176D - Codeforces(贪心+模拟)

Problem - 1176D - Codeforces

题目大意:假设给定一序列$a$,对于序列中的$a_i$,如果$a_i$是素数,那么找到第$a_i$个素数加入序列$a$的尾部,如果$a_i$是合数,那么将其最大的非$a_i$的因子加入序列$a$的尾部.现在给定一个长度为$2n$由长度为$n$的序列$a$生成的序列$b$(无序).求出原序列$a$.

解题思路:首先对于当前最大的素数,它一定由它次大的素数生成,对于当前最大的合数,它一定是原本存在于序列之中的,当前序列中存在它的最大因此,通过贪心规则确立了基本的构造原理之后,直接模拟即可.

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define syncfalse ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
const int MAX = 1e6 + 5;
const int N = 5e5 + 5;
int a[N];
bool notPrime[50000005];
int Prime[MAX];
int cnt = 0;
int n;
inline void solve(int n) {
    for (int i = 2; i <= n; ++i) {
        if (!notPrime[i]) {
            Prime[++cnt] = i;
        }
        for (int j = 1; j <= cnt && Prime[j] * i <= n; ++j) {
            notPrime[Prime[j] * i] = true;
            if (i % Prime[j] == 0)break;
        }
    }
}
int num[N];
int main() {
    syncfalse
#ifndef ONLINE_JUDGE
        freopen("in.txt", "r", stdin);
#endif
    solve(3000000);
    int n;
    cin >> n;
    n *= 2;
    vector<int>ans;
    for (int i = 1; i <= n; ++i) {
        cin >> a[i];
        num[a[i]]++;
    }
    sort(a + 1, a + 1 + n);
    for (int i = n; i >= 1; --i) {
        if (num[a[i]] == 0)continue;
        num[a[i]]--;
        if (notPrime[a[i]]) {
            for (int j = 1; j <= cnt; ++j) {
                if (a[i] % Prime[j] == 0) {
                    ll tem = a[i] / Prime[j];
                    num[tem]--;
                    ans.push_back(a[i]);
                    break;
                }
            }
        }
        else {
            ll index = lower_bound(Prime + 1, Prime + 1 + cnt, a[i]) - Prime;
            num[index]--;
            ans.push_back(index);
        }
    }
    for (auto x : ans)cout << x << " ";

    return 0;
}