結果
問題 | No.2377 SUM AND XOR on Tree |
ユーザー |
![]() |
提出日時 | 2023-07-07 21:47:35 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 258 ms / 4,000 ms |
コード長 | 1,207 bytes |
コンパイル時間 | 2,028 ms |
コンパイル使用メモリ | 180,392 KB |
実行使用メモリ | 17,824 KB |
最終ジャッジ日時 | 2024-07-21 17:36:27 |
合計ジャッジ時間 | 8,354 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 33 |
ソースコード
#include <bits/stdc++.h>using namespace std;const long long MOD = 998244353;int main(){int N;cin >> N;vector<vector<int>> E(N);for (int i = 0; i < N - 1; i++){int u, v;cin >> u >> v;u--;v--;E[u].push_back(v);E[v].push_back(u);}vector<int> p(N, -1);vector<vector<int>> c(N);vector<int> b;queue<int> Q;Q.push(0);while (!Q.empty()){int v = Q.front();Q.pop();b.push_back(v);for (int w : E[v]){if (w != p[v]){p[w] = v;c[v].push_back(w);Q.push(w);}}}reverse(b.begin(), b.end());vector<int> A(N);for (int i = 0; i < N; i++){cin >> A[i];}long long ans = 0;for (int i = 0; i < 30; i++){vector<vector<long long>> dp(2, vector<long long>(N, 0));for (int v : b){dp[A[v] >> i & 1][v] = 1;for (int w : c[v]){long long x = (dp[0][v] * dp[0][w] + dp[0][v] * dp[1][w] + dp[1][v] * dp[1][w]) % MOD;long long y = (dp[1][v] * dp[0][w] + dp[1][v] * dp[1][w] + dp[0][v] * dp[1][w]) % MOD;dp[0][v] = x;dp[1][v] = y;}}ans += (1 << i) * dp[1][0] % MOD;ans %= MOD;}cout << ans << endl;}