結果
問題 | No.2377 SUM AND XOR on Tree |
ユーザー |
|
提出日時 | 2023-07-12 02:57:18 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 346 ms / 4,000 ms |
コード長 | 1,718 bytes |
コンパイル時間 | 1,268 ms |
コンパイル使用メモリ | 133,476 KB |
最終ジャッジ日時 | 2025-02-15 09:55:13 |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 33 |
ソースコード
#include <iostream>#include <string>#include <vector>#include <algorithm>#include <functional>#include <cmath>#include <iomanip>#include <stack>#include <queue>#include <numeric>#include <map>#include <unordered_map>#include <set>#include <fstream>#include <chrono>#include <random>#include <bitset>//#include <atcoder/all>#define rep(i,n) for(int i=0;i<(n);i++)#define all(x) x.begin(), x.end()#define rall(x) x.rbegin(), x.rend()#define sz(x) ((int)(x).size())#define pb push_backusing ll = long long;using namespace std;template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }ll gcd(ll a, ll b) {return b?gcd(b,a%b):a;}ll lcm(ll a, ll b) {return a/gcd(a,b)*b;}const ll mod = 998244353;void dfs(int n, int p, vector<ll>& A, int s, vector<vector<int>>& G, vector<vector<ll>>& dp){rep(i,sz(G[n])){int x = G[n][i];if(x==p) continue;dfs(x,n,A,s,G,dp);}dp[0][n] = ((A[n]>>s)&1)==0;dp[1][n] = 1-dp[0][n];rep(i,sz(G[n])){int x = G[n][i];if(x==p) continue;ll t0 = dp[0][n];ll t1 = dp[1][n];dp[0][n] = (t0 * (dp[0][x] + dp[1][x]) + t1 * (dp[1][x])) % mod;dp[1][n] = (t0 * (dp[1][x]) + t1 * (dp[0][x] + dp[1][x])) % mod;}}int main(){int N; cin >> N;vector<vector<int>> G(N);rep(i,N-1){int a,b; cin >> a >> b; a--; b--;G[a].pb(b);G[b].pb(a);}vector<ll> A(N);rep(i,N) cin >> A[i];ll ans = 0;rep(s,30){vector<vector<ll>> dp(2,vector<ll>(N,0));dfs(0,-1,A,s,G,dp);ans += dp[1][0] * (1LL<<s) % mod;ans %= mod;}cout << ans % mod << '\n';return 0;};