結果
問題 | No.1843 Tree ANDistance |
ユーザー |
![]() |
提出日時 | 2022-02-18 21:27:23 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 467 ms / 2,000 ms |
コード長 | 939 bytes |
コンパイル時間 | 755 ms |
コンパイル使用メモリ | 82,276 KB |
実行使用メモリ | 9,112 KB |
最終ジャッジ日時 | 2024-06-29 08:18:15 |
合計ジャッジ時間 | 9,098 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 38 |
ソースコード
#include <vector> #include <iostream> #include <algorithm> #include <functional> using namespace std; int main() { int N; cin >> N; vector<int> A(N - 1), B(N - 1), C(N - 1); for (int i = 0; i < N - 1; i++) { cin >> A[i] >> B[i] >> C[i]; A[i] -= 1; B[i] -= 1; } const int MOD = 1000000007; int answer = 0; for (int i = 0; i < 30; i++) { vector<vector<int> > G(N); for (int j = 0; j < N - 1; j++) { if ((C[j] >> i) & 1) { G[A[j]].push_back(B[j]); G[B[j]].push_back(A[j]); } } vector<bool> vis(N, false); function<int(int)> dfs = [&](int pos) { int res = 1; vis[pos] = true; for (int j : G[pos]) { if (!vis[j]) { res += dfs(j); } } return res; }; for (int j = 0; j < N; j++) { if (!vis[j]) { int res = dfs(j); long long combs = (1LL * res * (res - 1) / 2) % MOD; answer = (answer + (combs << i)) % MOD; } } } cout << answer << endl; return 0; }