結果

問題 No.2200 Weird Shortest Path
ユーザー 鴨志田卓
提出日時 2023-03-11 01:31:37
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 247 ms / 2,000 ms
コード長 696 bytes
コンパイル時間 2,023 ms
コンパイル使用メモリ 202,232 KB
最終ジャッジ日時 2025-02-11 09:42:10
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using edge = tuple<int, int, int>;
const int N = 2e5 + 10;

int fa[N];

int find(int x) {
    return fa[x] > 0 ? find(fa[x]) : x;
}

int n, m;

vector<edge> e;

int main() {
    memset(fa, -1, sizeof fa);
    cin >> n >> m;
    for(int i = 1, u, v, w; i <= m; i ++) {
        cin >> u >> v >> w;
        e.push_back({w, u, v});
    }
    sort(e.begin(), e.end());
    long long ans = 0;
    for(auto &[w, u, v] : e) {
        int fu = find(u), fv = find(v);
        if(fu == fv) continue;
        ans += (long long)w * fa[fu] * fa[fv];
        if(fa[fu] < fa[fv]) swap(fu, fv);
        fa[fv] += fa[fu];
        fa[fu] = fv;
    }
    cout << ans;
}
0