結果
問題 | No.1843 Tree ANDistance |
ユーザー | otoshigo |
提出日時 | 2022-02-18 21:45:03 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 400 ms / 2,000 ms |
コード長 | 1,982 bytes |
コンパイル時間 | 2,170 ms |
コンパイル使用メモリ | 208,956 KB |
実行使用メモリ | 6,288 KB |
最終ジャッジ日時 | 2023-09-11 18:50:35 |
合計ジャッジ時間 | 11,058 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 341 ms
6,256 KB |
testcase_01 | AC | 341 ms
6,244 KB |
testcase_02 | AC | 335 ms
6,164 KB |
testcase_03 | AC | 335 ms
6,264 KB |
testcase_04 | AC | 341 ms
6,252 KB |
testcase_05 | AC | 320 ms
6,204 KB |
testcase_06 | AC | 336 ms
6,152 KB |
testcase_07 | AC | 322 ms
6,148 KB |
testcase_08 | AC | 332 ms
6,288 KB |
testcase_09 | AC | 320 ms
6,000 KB |
testcase_10 | AC | 311 ms
6,012 KB |
testcase_11 | AC | 340 ms
6,156 KB |
testcase_12 | AC | 292 ms
5,904 KB |
testcase_13 | AC | 313 ms
5,996 KB |
testcase_14 | AC | 12 ms
4,376 KB |
testcase_15 | AC | 13 ms
4,380 KB |
testcase_16 | AC | 7 ms
4,380 KB |
testcase_17 | AC | 6 ms
4,384 KB |
testcase_18 | AC | 3 ms
4,380 KB |
testcase_19 | AC | 15 ms
4,376 KB |
testcase_20 | AC | 8 ms
4,376 KB |
testcase_21 | AC | 2 ms
4,376 KB |
testcase_22 | AC | 2 ms
4,380 KB |
testcase_23 | AC | 1 ms
4,376 KB |
testcase_24 | AC | 2 ms
4,376 KB |
testcase_25 | AC | 2 ms
4,376 KB |
testcase_26 | AC | 2 ms
4,380 KB |
testcase_27 | AC | 1 ms
4,376 KB |
testcase_28 | AC | 367 ms
5,616 KB |
testcase_29 | AC | 246 ms
4,924 KB |
testcase_30 | AC | 255 ms
4,892 KB |
testcase_31 | AC | 400 ms
5,832 KB |
testcase_32 | AC | 375 ms
5,512 KB |
testcase_33 | AC | 128 ms
4,380 KB |
testcase_34 | AC | 295 ms
4,992 KB |
testcase_35 | AC | 2 ms
4,376 KB |
testcase_36 | AC | 2 ms
4,380 KB |
testcase_37 | AC | 1 ms
4,376 KB |
ソースコード
#include <bits/stdc++.h> #include <atcoder/modint> using namespace std; using namespace atcoder; using ll = long long; using mint = modint1000000007; #define rep(i,n) for (int i = 0; i < n; i++) #define rrep(i,n) for (int i = n-1; i >= 0; i--) #define rep2(i,a,b) for (int i = a; i < b; i++) #define rrep2(i,a,b) for (int i = a-1; i >= b; i--) #define rep3(i,a,b,c) for (int i = a; i < b; i+=c) #define rrep3(i,a,b,c) for (int i = a-1; i >= b; i-=c) #define all(v) v.begin(),v.end() #define rall(v) v.rbegin(),v.rend() template<class T> bool chmax(T &a, T b){if (a < b){a = b;return true;} else return false;} template<class T> bool chmin(T &a, T b){if (a > b){a = b;return true;} else return false;} template<class T> T pow(T x, ll y){if (y == 0) return 1.0;T p = pow(x,y/2);if (y%2 == 0) return p*p;else return p*p*x;} struct UnionFind { vector<int> par, siz; UnionFind(int n) : par(n,-1), siz(n,1) {} int find(int x){ if (par[x] == -1) return x; else{ par[x] = find(par[x]); return par[x]; } } bool same(int x, int y){ return find(x) == find(y); } int size(int x){ return siz[find(x)]; } void unite(int x, int y){ x = find(x); y = find(y); if (x == y) return; if (siz[x] < siz[y]) swap(x,y); siz[x] += siz[y]; par[y] = x; } }; int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); int n; cin >> n; vector<ll> A(n-1), B(n-1), C(n-1); rep(i,n-1){ cin >> A[i] >> B[i] >> C[i]; --A[i]; --B[i]; } mint ans = 0; rep(i,30){ UnionFind uni(n); rep(j,n-1){ if (C[j]>>i&1) uni.unite(A[j],B[j]); } vector<bool> flag(n); rep(j,n){ if (flag[uni.find(j)]) continue; flag[uni.find(j)] = true; ll l = uni.size(j); ans += mint(2).pow(i)*l*(l-1)/2; } } cout << ans.val() << endl; }