結果
問題 | No.2377 SUM AND XOR on Tree |
ユーザー | srjywrdnprkt |
提出日時 | 2023-07-09 02:17:08 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 759 ms / 4,000 ms |
コード長 | 2,744 bytes |
コンパイル時間 | 2,502 ms |
コンパイル使用メモリ | 214,064 KB |
実行使用メモリ | 24,164 KB |
最終ジャッジ日時 | 2023-09-30 04:40:11 |
合計ジャッジ時間 | 15,712 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,380 KB |
testcase_01 | AC | 1 ms
4,380 KB |
testcase_02 | AC | 1 ms
4,380 KB |
testcase_03 | AC | 1 ms
4,380 KB |
testcase_04 | AC | 1 ms
4,376 KB |
testcase_05 | AC | 1 ms
4,380 KB |
testcase_06 | AC | 2 ms
4,380 KB |
testcase_07 | AC | 2 ms
4,376 KB |
testcase_08 | AC | 732 ms
14,940 KB |
testcase_09 | AC | 759 ms
14,924 KB |
testcase_10 | AC | 751 ms
14,872 KB |
testcase_11 | AC | 720 ms
14,900 KB |
testcase_12 | AC | 726 ms
14,860 KB |
testcase_13 | AC | 670 ms
15,052 KB |
testcase_14 | AC | 665 ms
14,348 KB |
testcase_15 | AC | 691 ms
14,732 KB |
testcase_16 | AC | 633 ms
14,352 KB |
testcase_17 | AC | 700 ms
14,636 KB |
testcase_18 | AC | 2 ms
4,376 KB |
testcase_19 | AC | 2 ms
4,376 KB |
testcase_20 | AC | 2 ms
4,376 KB |
testcase_21 | AC | 2 ms
4,376 KB |
testcase_22 | AC | 2 ms
4,380 KB |
testcase_23 | AC | 670 ms
14,872 KB |
testcase_24 | AC | 633 ms
14,912 KB |
testcase_25 | AC | 723 ms
14,804 KB |
testcase_26 | AC | 231 ms
24,164 KB |
testcase_27 | AC | 236 ms
24,124 KB |
testcase_28 | AC | 236 ms
24,116 KB |
testcase_29 | AC | 175 ms
15,148 KB |
testcase_30 | AC | 173 ms
15,204 KB |
testcase_31 | AC | 175 ms
15,196 KB |
testcase_32 | AC | 206 ms
15,016 KB |
testcase_33 | AC | 204 ms
14,884 KB |
testcase_34 | AC | 211 ms
14,608 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; const ll modc = 998244353; class mint { ll x; public: mint(ll x=0) : x((x%modc+modc)%modc) {} mint operator-() const { return mint(-x); } mint& operator+=(const mint& a) { if ((x += a.x) >= modc) x -= modc; return *this; } mint& operator-=(const mint& a) { if ((x += modc-a.x) >= modc) x -= modc; return *this; } mint& operator*=(const mint& a) { (x *= a.x) %= modc; return *this; } mint operator+(const mint& a) const { mint res(*this); return res+=a; } mint operator-(const mint& a) const { mint res(*this); return res-=a; } mint operator*(const mint& a) const { mint res(*this); return res*=a; } mint pow(ll t) const { if (!t) return 1; mint a = pow(t>>1); a *= a; if (t&1) a *= *this; return a; } mint inv() const { return pow(modc-2); } mint& operator/=(const mint& a) { return (*this) *= a.inv(); } mint operator/(const mint& a) const { mint res(*this); return res/=a; } bool operator == (const mint& a) const{ return x == a.x; } friend ostream& operator<<(ostream& os, const mint& m){ os << m.x; return os; } friend istream& operator>>(istream& ip, mint &m) { ll t; ip >> t; m = mint(t); return ip; } }; vector<vector<int>> E; vector<int> bt; vector<vector<mint>> dp; void dfs(int from, int p){ mint z=0, o=0; if (bt[from] == 0) z = 1; else o = 1; for (auto to : E[from]){ if (p == to) continue; dfs(to, from); mint zz=0, oo=0; //つなげる zz += z * dp[to][0]; zz += o * dp[to][1]; oo += z * dp[to][1]; oo += o * dp[to][0]; //削除する zz += z * dp[to][1]; oo += o * dp[to][1]; swap(o, oo); swap(z, zz); } dp[from] = {z, o}; } int main(){ int N, x, y; cin >> N; E.resize(N); for (int i=0; i<N-1; i++){ cin >> x >> y; x--; y--; E[x].push_back(y); E[y].push_back(x); } vector<int> A(N); for (int i=0; i<N; i++) cin >> A[i]; mint ans=0; dp.resize(N, vector<mint>(2)); bt.resize(N); for (int i=0; i<30; i++){ for (int j=0; j<N; j++){ dp[j][0] = 0; dp[j][1] = 0; } for (int j=0; j<N; j++){ if (1<<i & A[j]) bt[j] = 1; else bt[j] = 0; } dfs(0, -1); ans += dp[0][1] * (1<<i); } cout << ans << endl; return 0; }