結果
問題 | No.2377 SUM AND XOR on Tree |
ユーザー |
|
提出日時 | 2023-07-07 21:51:27 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 221 ms / 4,000 ms |
コード長 | 789 bytes |
コンパイル時間 | 744 ms |
コンパイル使用メモリ | 72,896 KB |
実行使用メモリ | 18,916 KB |
最終ジャッジ日時 | 2024-07-21 17:43:50 |
合計ジャッジ時間 | 5,821 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 33 |
ソースコード
#include<iostream>#include<vector>#include<cassert>#include<atcoder/modint>using namespace std;using mint=atcoder::modint998244353;int N,A[1<<17];vector<int>G[1<<17];mint dp[1<<17][2];void dfs(int u,int p,int k){dp[u][0]=dp[u][1]=0;dp[u][A[u]>>k&1]=1;for(int v:G[u])if(v!=p){dfs(v,u,k);mint dp0=dp[u][0]*dp[v][1]+dp[u][0]*dp[v][0]+dp[u][1]*dp[v][1];mint dp1=dp[u][1]*dp[v][1]+dp[u][0]*dp[v][1]+dp[u][1]*dp[v][0];dp[u][0]=dp0;dp[u][1]=dp1;}}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);cin>>N;for(int i=1;i<N;i++){int u,v;cin>>u>>v;u--,v--;G[u].push_back(v);G[v].push_back(u);}for(int i=0;i<N;i++)cin>>A[i];mint ans=0;for(int k=30;k--;){ans*=2;dfs(0,-1,k);ans+=dp[0][1];}cout<<ans.val()<<"\n";}