結果

問題 No.1843 Tree ANDistance
ユーザー Lê Hoàng NgôLê Hoàng Ngô
提出日時 2024-06-28 23:02:46
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,064 ms / 2,000 ms
コード長 2,253 bytes
コンパイル時間 1,759 ms
コンパイル使用メモリ 176,680 KB
実行使用メモリ 32,256 KB
最終ジャッジ日時 2024-06-28 23:03:08
合計ジャッジ時間 19,830 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,045 ms
32,256 KB
testcase_01 AC 1,012 ms
32,128 KB
testcase_02 AC 1,064 ms
32,128 KB
testcase_03 AC 1,049 ms
32,256 KB
testcase_04 AC 1,026 ms
32,128 KB
testcase_05 AC 565 ms
32,128 KB
testcase_06 AC 577 ms
32,128 KB
testcase_07 AC 990 ms
30,592 KB
testcase_08 AC 969 ms
31,616 KB
testcase_09 AC 939 ms
30,592 KB
testcase_10 AC 941 ms
30,208 KB
testcase_11 AC 1,041 ms
31,872 KB
testcase_12 AC 483 ms
29,568 KB
testcase_13 AC 539 ms
29,696 KB
testcase_14 AC 17 ms
5,376 KB
testcase_15 AC 19 ms
5,376 KB
testcase_16 AC 8 ms
5,376 KB
testcase_17 AC 8 ms
5,376 KB
testcase_18 AC 4 ms
5,376 KB
testcase_19 AC 15 ms
5,376 KB
testcase_20 AC 8 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 1 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 378 ms
25,600 KB
testcase_29 AC 238 ms
18,560 KB
testcase_30 AC 322 ms
18,688 KB
testcase_31 AC 422 ms
27,648 KB
testcase_32 AC 385 ms
26,240 KB
testcase_33 AC 129 ms
11,008 KB
testcase_34 AC 338 ms
21,376 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>
using namespace std;
#define ll long long
#define all(x) x.begin(),x.end()
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define rep_r(i,a,b) for(int i=a;i>=b;i--)
#define each(a,x) for (auto& x : a)
using pi = pair<int,int>;
using pl = pair<ll,ll>;
using vi = vector<int>;
using vl = vector<ll>;
#define sz(x) int(x.size())
#define so(x) sort(all(x))
#define so_r(x) sort(all(x),greater<int>())
#define lb lower_bound
#define ub upper_bound
const char nl = '\n';
int dx[4] = {1,-1,0,0};
int dy[4] = {0,0,1,-1};
int bit_cnt(int x){
    return __builtin_popcount(x);
}
ll bex(ll a, ll b, ll mod = 1e9 + 7){ll res = 1LL; while(b){ if (b&1) res = res * a % mod; a = a * a % mod; b >>= 1;} return res;}
template<class t,class u> bool chmax(t&a,u b){if(a<b){a=b;return true;}else return false;}
template<class t,class u> bool chmin(t&a,u b){if(b<a){a=b;return true;}else return false;}

const int MOD = 1e9 + 7;
const int N = 5e5 + 5;
const ll INF = 1e16;
struct DSU{
    vector<int> fa,sz;
    void init(int n){ 
        fa.resize(n+1);
        sz.resize(n+1);
        rep(i,1,n) fa[i] = i;
        rep(i,1,n) sz[i] = 1;
    }
    int find(int u) {
        return (u == fa[u] ? u : fa[u] = find(fa[u]));
    }
    void merge(int u, int v){ 
        u = find(u);
        v = find(v); 
        if (u == v) return;
        if (sz[u] < sz[v]) swap(u,v);
        fa[v] = u;
        sz[u] += sz[v];
    }
};
DSU dsu[31];
int n;
ll calc(int j, int x){
    return 1LL * x * (x - 1) / 2 % MOD * (1LL << j) % MOD;
}
void solve(){
    cin >> n;
    rep(i,0,30) dsu[i].init(n);
    rep(i,1,n-1){
        int a,b,c; cin >> a >> b >> c;
        for (int j = 0; j <= 30; j++){
            if (c>>j&1){
                dsu[j].merge(a,b);
            }
        }
    }
    ll ans = 0;
    rep(i,0,30){
        set<pair<int,int>> st;
        rep(j,1,n){
            int fa = dsu[i].find(j);
            st.insert({fa,dsu[i].sz[fa]});
        }
        each(st,p){
            int comp_size = p.second;
            ans += calc(i, comp_size);
            ans %= MOD;
        }   
    }
    cout << ans << nl;
}

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int t = 1; 
    // cin >> t;

    while (t--){
        solve();
    }
}
0