結果

問題 No.1843 Tree ANDistance
ユーザー ShirotsumeShirotsume
提出日時 2022-02-18 22:06:29
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,105 ms / 2,000 ms
コード長 976 bytes
コンパイル時間 4,396 ms
コンパイル使用メモリ 270,400 KB
実行使用メモリ 13,904 KB
最終ジャッジ日時 2024-06-29 09:00:26
合計ジャッジ時間 24,875 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,029 ms
13,904 KB
testcase_01 AC 1,019 ms
13,788 KB
testcase_02 AC 1,006 ms
13,868 KB
testcase_03 AC 1,005 ms
13,896 KB
testcase_04 AC 1,029 ms
13,900 KB
testcase_05 AC 820 ms
13,776 KB
testcase_06 AC 822 ms
13,768 KB
testcase_07 AC 950 ms
13,388 KB
testcase_08 AC 971 ms
13,568 KB
testcase_09 AC 952 ms
13,368 KB
testcase_10 AC 927 ms
13,088 KB
testcase_11 AC 996 ms
13,700 KB
testcase_12 AC 713 ms
12,944 KB
testcase_13 AC 747 ms
13,004 KB
testcase_14 AC 26 ms
5,376 KB
testcase_15 AC 28 ms
5,376 KB
testcase_16 AC 13 ms
5,376 KB
testcase_17 AC 11 ms
5,376 KB
testcase_18 AC 5 ms
5,376 KB
testcase_19 AC 25 ms
5,376 KB
testcase_20 AC 13 ms
5,376 KB
testcase_21 AC 1 ms
5,376 KB
testcase_22 AC 1 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 1 ms
5,376 KB
testcase_27 AC 1 ms
5,376 KB
testcase_28 AC 942 ms
11,556 KB
testcase_29 AC 607 ms
8,892 KB
testcase_30 AC 651 ms
9,008 KB
testcase_31 AC 1,105 ms
12,224 KB
testcase_32 AC 993 ms
11,712 KB
testcase_33 AC 292 ms
6,144 KB
testcase_34 AC 775 ms
9,808 KB
testcase_35 AC 2 ms
5,376 KB
testcase_36 AC 2 ms
5,376 KB
testcase_37 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll = long long;
#define all(x) x.begin(), x.end()
#define rep(i, n) for (int i = 0; i < n; i++)
#define mint modint1000000007
int main(void)
{
    int n;
    cin >> n;
    mint ans, bi2;
    bi2 = 1;
    vector<vector<int>> abc(n, vector<int>(3));
    rep(i, n - 1)rep(j, 3){
        cin >> abc[i][j];
    }
    rep(bit, 32){
        dsu U(n + 1);
        rep(i, n - 1){
            int a = abc[i][0];
            int b = abc[i][1]; 
            int c = abc[i][2];
            if(1 & (c >> bit)) U.merge(a - 1, b - 1);
        }
        map<int, int> s;
        rep(i, n){
            int x = U.leader(i);
            if (s.find(x) != s.end()) continue;
            else{
                s[x] = 1;
                mint c = U.size(i);
                ans += bi2 * c * (c - 1) / 2;
            }
        }
        bi2 *= 2;

    }
    cout << ans.val() << endl;
    return 0;
}
0