結果

問題 No.1917 LCMST
ユーザー Dispersion-atc
提出日時 2024-03-01 02:43:21
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,450 bytes
コンパイル時間 2,330 ms
コンパイル使用メモリ 202,564 KB
最終ジャッジ日時 2025-02-19 22:08:03
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 14 WA * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

template<class T>
bool chmin(T &a, const T &b) {if (a > b) {a = b; return 1;} return 0;}

using ll = long long;
const ll amax = 1e5 + 100;

ll gcd(ll a, ll b)
{
    if (b == 0) return a;
    else return gcd(b, a % b);
}

ll lcm(ll a, ll b)
{
    return a / gcd(a, b) * b;
}

int main()
{
    ll N; cin >> N;
    vector<ll> A(N);
    for (auto &a : A) cin >> a;
    
    vector<ll> uniq;
    vector<ll> cnt(amax, 0);
    ll res = 0;
    for (auto a : A)
    {
        if (cnt[a] == 0) uniq.emplace_back(a);
        else res += a;
        cnt[a] += 1;
    }
    
    sort(uniq.begin(), uniq.end());
    vector<vector<ll>> vec(amax);
    const ll INF = 1e18;
    for (auto a : uniq)
    {
        vector<ll> divisor;
        for (int i = 1; i * i <= a; ++i)
        {
            if (a % i == 0)
            {
                divisor.emplace_back(i);
                if (a / i != i) divisor.emplace_back(a / i);
            }
        }
        
        ll mn = INF;
        for (auto d : divisor)
        {
            if (vec[d].size() == 0)
            {
                vec[d].emplace_back(a);
                continue;
            }
            ll t = vec[d][0];
            chmin(mn, lcm(a, t));
            
            vec[d].emplace_back(a);
        }
        // cout << a << " " << mn << endl;
        if (mn == INF) continue;
        res += mn;
    }
    cout << res << endl;
    
    return 0;
}
0