mayoko’s diary

プロコンとかいろいろ。

yukicoder No.368 LCM of K-products

解法

最小公倍数は, 各素因数が最大のものを組み合わせたものになります。

よって, 各素因数を調べて, p の素因数を考える場合は p の指数が多い順に K 個取ってくる, というようにやれば OK です。

const int MAXN = 1010;
int A[MAXN];
map<int, vi> mp;

void calc(int x, map<int, int>& M) {
    for (int i = 2; i*i <= x; i++) {
            int cnt = 0;
            while (x%i == 0) {
                        x /= i;
                        cnt++;
                    }
            if (cnt) M[i] += cnt;
        }
    if (x > 1) M[x] += 1;
}

const int MOD = 1e9+7;

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    int N, K;
    cin >> N >> K;
    for (int i = 0; i < N; i++) {
        cin >> A[i];
        map<int, int> M;
        calc(A[i], M);
        for (auto p : M) {
            mp[p.first].push_back(p.second);
        }
    }
    ll ans = 1;
    for (auto& p : mp) {
        sort(p.second.rbegin(), p.second.rend());
        int sz = min<int>(K, p.second.size());
        int sum = 0;
        for (int i = 0; i < sz; i++) sum += p.second[i];
        for (int i = 0; i < sum; i++) (ans *= p.first) %= MOD;
    }
    cout << ans << endl;
    return 0;
}