結果

問題 No.1917 LCMST
ユーザー Trí Tâm NguyễnTrí Tâm Nguyễn
提出日時 2023-02-11 18:17:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 600 ms / 4,000 ms
コード長 1,987 bytes
コンパイル時間 5,818 ms
コンパイル使用メモリ 212,412 KB
実行使用メモリ 138,880 KB
最終ジャッジ日時 2024-07-08 05:54:08
合計ジャッジ時間 24,300 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 86 ms
58,420 KB
testcase_01 AC 85 ms
59,416 KB
testcase_02 AC 90 ms
59,320 KB
testcase_03 AC 91 ms
59,456 KB
testcase_04 AC 90 ms
59,888 KB
testcase_05 AC 90 ms
58,908 KB
testcase_06 AC 88 ms
58,528 KB
testcase_07 AC 89 ms
58,984 KB
testcase_08 AC 89 ms
59,472 KB
testcase_09 AC 586 ms
138,880 KB
testcase_10 AC 579 ms
138,588 KB
testcase_11 AC 600 ms
137,236 KB
testcase_12 AC 561 ms
132,704 KB
testcase_13 AC 375 ms
125,320 KB
testcase_14 AC 583 ms
135,436 KB
testcase_15 AC 321 ms
123,304 KB
testcase_16 AC 386 ms
127,152 KB
testcase_17 AC 461 ms
129,540 KB
testcase_18 AC 489 ms
129,196 KB
testcase_19 AC 340 ms
121,236 KB
testcase_20 AC 309 ms
120,212 KB
testcase_21 AC 364 ms
120,904 KB
testcase_22 AC 421 ms
121,228 KB
testcase_23 AC 317 ms
120,956 KB
testcase_24 AC 339 ms
122,200 KB
testcase_25 AC 310 ms
120,528 KB
testcase_26 AC 351 ms
122,960 KB
testcase_27 AC 334 ms
123,528 KB
testcase_28 AC 339 ms
123,452 KB
testcase_29 AC 551 ms
137,728 KB
testcase_30 AC 552 ms
137,956 KB
testcase_31 AC 569 ms
138,440 KB
testcase_32 AC 551 ms
137,944 KB
testcase_33 AC 552 ms
137,684 KB
testcase_34 AC 225 ms
103,076 KB
testcase_35 AC 220 ms
103,520 KB
testcase_36 AC 216 ms
103,748 KB
testcase_37 AC 563 ms
138,804 KB
testcase_38 AC 564 ms
138,724 KB
testcase_39 AC 577 ms
137,760 KB
testcase_40 AC 559 ms
138,028 KB
testcase_41 AC 568 ms
138,880 KB
testcase_42 AC 579 ms
138,300 KB
testcase_43 AC 220 ms
103,364 KB
testcase_44 AC 225 ms
103,704 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// Make the best become better
// No room for laziness
#include<bits/stdc++.h>

#define int long long
#define pb push_back
#define fi first
#define se second
using namespace std;
using ll = long long;
using ld = long double;
using ull = unsigned long long;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const int maxN = 1e6 + 5;
const int mod = 1e9 + 7;
const ll oo = 1e18;
int n;
int a[maxN];
vector<int> vc[maxN];
vector<int> L[maxN];
int lab[maxN];
int Findset(int u)
{
    return lab[u] < 0 ? u : lab[u] = Findset(lab[u]);
}
bool Unite(int u, int v)
{
    int r = Findset(u), s = Findset(v);
    if(r == s) return false;
    if(lab[r] > lab[s]) swap(r, s);
    lab[r] += lab[s];
    lab[s] = r;
    return true;
}
struct TEdge
{
    int u, v, w;
};
vector<TEdge> e;
void ReadInput()
{
    cin >> n;
    for(int i=1; i<=n; i++)
    {
        cin >> a[i];
        vc[a[i]].pb(i);
    }
}
void Solve()
{
    for(int i=1; i<maxN; i++)
    {
        for(int j=1; j<vc[i].size(); j++)
            e.pb({vc[i][0], vc[i][j], i});
        while(vc[i].size() > 1) vc[i].pop_back();
    }
    for(int i=1; i<maxN; i++)
    {
        for(int j=i; j<maxN; j+=i)
        {
            for(int v : vc[j])
                L[i].pb(v);
        }
    }
    for(int i=1; i<maxN; i++)
    {
        sort(L[i].begin(), L[i].end(), [](int i, int j)
        {
            return a[i] < a[j];
        });
        for(int j=1; j<L[i].size(); j++)
            e.pb({L[i][0], L[i][j], a[L[i][0]] * a[L[i][j]] / __gcd(a[L[i][0]], a[L[i][j]])});
    }
    memset(lab, -1, sizeof lab);
    sort(e.begin(), e.end(), [](TEdge p, TEdge q)
    {
        return p.w < q.w;
    });
    int res = 0;
    for(TEdge a : e)
    {
        //cout << a.u << " " << a.v << " " << a.w << '\n';
        if(Unite(a.u, a.v)) res += a.w;
    }
    cout << res;
}
int32_t main()
{
    //freopen("sol.inp", "r", stdin);
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    ReadInput();
    Solve();
}
0