結果

問題 No.1843 Tree ANDistance
ユーザー bit_kyopro
提出日時 2022-02-18 22:22:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 397 ms / 2,000 ms
コード長 3,504 bytes
コンパイル時間 2,189 ms
コンパイル使用メモリ 207,256 KB
最終ジャッジ日時 2025-01-28 00:18:13
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

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