結果

問題 No.1917 LCMST
ユーザー Trí Tâm NguyễnTrí Tâm Nguyễn
提出日時 2024-03-12 20:44:44
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 630 ms / 4,000 ms
コード長 1,862 bytes
コンパイル時間 2,058 ms
コンパイル使用メモリ 178,884 KB
実行使用メモリ 137,828 KB
最終ジャッジ日時 2024-09-29 22:25:49
合計ジャッジ時間 25,798 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 114 ms
58,188 KB
testcase_01 AC 123 ms
58,064 KB
testcase_02 AC 116 ms
58,188 KB
testcase_03 AC 115 ms
58,576 KB
testcase_04 AC 115 ms
58,572 KB
testcase_05 AC 118 ms
58,568 KB
testcase_06 AC 121 ms
58,828 KB
testcase_07 AC 113 ms
58,576 KB
testcase_08 AC 108 ms
58,184 KB
testcase_09 AC 575 ms
136,916 KB
testcase_10 AC 618 ms
136,892 KB
testcase_11 AC 573 ms
136,468 KB
testcase_12 AC 547 ms
131,960 KB
testcase_13 AC 425 ms
124,500 KB
testcase_14 AC 555 ms
134,216 KB
testcase_15 AC 358 ms
120,940 KB
testcase_16 AC 404 ms
124,672 KB
testcase_17 AC 455 ms
127,428 KB
testcase_18 AC 503 ms
128,428 KB
testcase_19 AC 368 ms
120,756 KB
testcase_20 AC 349 ms
119,328 KB
testcase_21 AC 376 ms
120,500 KB
testcase_22 AC 353 ms
119,548 KB
testcase_23 AC 351 ms
119,220 KB
testcase_24 AC 369 ms
120,488 KB
testcase_25 AC 355 ms
119,892 KB
testcase_26 AC 367 ms
122,052 KB
testcase_27 AC 368 ms
122,240 KB
testcase_28 AC 376 ms
121,484 KB
testcase_29 AC 594 ms
137,704 KB
testcase_30 AC 611 ms
137,572 KB
testcase_31 AC 594 ms
137,480 KB
testcase_32 AC 610 ms
137,644 KB
testcase_33 AC 605 ms
137,644 KB
testcase_34 AC 247 ms
103,196 KB
testcase_35 AC 277 ms
103,320 KB
testcase_36 AC 281 ms
103,192 KB
testcase_37 AC 617 ms
137,652 KB
testcase_38 AC 626 ms
137,808 KB
testcase_39 AC 630 ms
137,608 KB
testcase_40 AC 613 ms
137,728 KB
testcase_41 AC 621 ms
137,828 KB
testcase_42 AC 612 ms
137,440 KB
testcase_43 AC 273 ms
103,340 KB
testcase_44 AC 284 ms
103,348 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]] / i});
    }
    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)
    {
        if(Unite(a.u, a.v)) res += a.w;
    }
    cout << res;
}
int32_t main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    ReadInput();
    Solve();
}
0