結果

問題 No.1843 Tree ANDistance
ユーザー bit_kyoprobit_kyopro
提出日時 2022-02-18 22:22:58
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 217 ms / 2,000 ms
コード長 3,504 bytes
コンパイル時間 2,349 ms
コンパイル使用メモリ 213,428 KB
実行使用メモリ 15,688 KB
最終ジャッジ日時 2024-06-29 09:14:27
合計ジャッジ時間 7,371 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 211 ms
14,860 KB
testcase_01 AC 212 ms
14,736 KB
testcase_02 AC 215 ms
14,824 KB
testcase_03 AC 211 ms
14,824 KB
testcase_04 AC 213 ms
15,688 KB
testcase_05 AC 179 ms
14,332 KB
testcase_06 AC 198 ms
14,732 KB
testcase_07 AC 203 ms
14,192 KB
testcase_08 AC 205 ms
14,404 KB
testcase_09 AC 197 ms
14,068 KB
testcase_10 AC 197 ms
13,896 KB
testcase_11 AC 217 ms
15,280 KB
testcase_12 AC 165 ms
13,232 KB
testcase_13 AC 184 ms
13,376 KB
testcase_14 AC 8 ms
5,376 KB
testcase_15 AC 8 ms
5,376 KB
testcase_16 AC 5 ms
5,376 KB
testcase_17 AC 4 ms
5,376 KB
testcase_18 AC 3 ms
5,376 KB
testcase_19 AC 9 ms
5,376 KB
testcase_20 AC 5 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 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 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 109 ms
12,128 KB
testcase_29 AC 74 ms
9,196 KB
testcase_30 AC 76 ms
9,380 KB
testcase_31 AC 126 ms
12,936 KB
testcase_32 AC 117 ms
12,896 KB
testcase_33 AC 34 ms
6,092 KB
testcase_34 AC 84 ms
10,136 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 atcoder;
//using mint = modint1000000007;
using mint = modint998244353;
*/
//g++ -I/opt/ac-library ./**.cpp
using namespace std;
using ll = long long;
using ull = unsigned long long;
const long long MOD = 1000000007;

const long double PI = 3.14159265358979;
const long long INF = 1LL<<60;
template <typename T> bool chmax(T &a, const T& b){if(a < b){a = b;return true;}return false;}
template <typename T> bool chmin(T &a, const T& b){if(a > b){a = b;return true;}return false;}
#define deb(var) do{cout << #var << " : "; view(var);}while(0)
template<typename T> void view(T e){cout << e << endl;}
void view(vector<string>& v){cout << endl;for(auto& s :v){view(s);}cout << endl;}
template<typename T> void view(vector<T>& v){for(auto& e :v){cout << e << " ";}cout << endl;}
template<typename T> void view(vector<vector<T>>& vv){cout << endl;for(auto& v:vv){view(v);}}
ll gcd(ll a, ll b){if (b == 0) return a;else return gcd(b, a % b);}
ll lcm(ll x,ll y){return ll(x/gcd(x,y))*y;}
template<typename T> using min_priority_queue = priority_queue<T, vector<T>, greater<T>>;
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define int long long


//a^n(int は ll)
ll mypow(ll a, ll n, ll mod) {
    ll res = 1;
    a %= mod;
    while (n) {
        if (n & 1) {
            res = res * a % mod;
        }
        a = a * a % mod;
        n >>= 1;
    }
    return res;
}
int com(int n) {
    return (n * (n - 1) / 2) % MOD;
}

struct UnionFind {
    vector<int> par, siz;

    // 初期化
    UnionFind(int n) : par(n, -1) , siz(n, 1) { }
    
    // 根を求める
    int root(int x) {
        if (par[x] == -1) return x; // x が根の場合は x を返す
        else return par[x] = root(par[x]);
    }

    // x と y が同じグループに属するかどうか (根が一致するかどうか)
    bool issame(int x, int y) {
        return root(x) == root(y);
    }

    // x を含むグループと y を含むグループとを併合する
    bool unite(int x, int y) {
        // x, y をそれぞれ根まで移動する
        x = root(x);
        y = root(y);

        // すでに同じグループのときは何もしない
        if (x == y) return false; 

        // union by size (y 側のサイズが小さくなるようにする)
        if (siz[x] < siz[y]) swap(x, y);

        // y を x の子とする
        par[y] = x;
        siz[x] += siz[y];
        return true;
    }

    // x を含むグループのサイズ
    int size(int x) {
        return siz[root(x)];
    }
};

int32_t main() {
    int n; cin >> n;
    vector<vector<pair<int, int>>> G(n);
    vector<pair<pair<int,int>, int>> edges;
    for (int i=0; i<n-1; i++) {
        int a, b, c; cin >> a >> b >> c;
        a--; b--;
        G[a].push_back({b, c});
        G[b].push_back({a, c});
        edges.push_back({{a, b}, c});
    }

    int ans = 0;
    for (int d=0; d<40; d++) {
        UnionFind uf(n);
        int res = 0;
        for (int i=0; i<n - 1; i++) {
            auto [a, b] = edges[i].first;
            int c = edges[i].second;
            if (c & 1LL<<d) {
                uf.unite(a, b);
            }
        }
        for (int i=0; i<n; i++) {
            if ( uf.root(i) == i ) {
                int l = uf.size(i);
                res += com(l);
                res %= MOD;
            }
        }
        ans += res * mypow(2, d, MOD);
        ans %= MOD;
    }
    cout << ans << endl;
}
0