結果
問題 | No.2214 Products on Tree |
ユーザー |
![]() |
提出日時 | 2023-02-10 21:53:33 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 230 ms / 3,000 ms |
コード長 | 1,165 bytes |
コンパイル時間 | 1,864 ms |
コンパイル使用メモリ | 180,884 KB |
実行使用メモリ | 30,772 KB |
最終ジャッジ日時 | 2024-07-07 16:06:44 |
合計ジャッジ時間 | 7,333 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 |
ソースコード
#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 A, B; cin >> A >> B; A--; B--; E[A].push_back(B); E[B].push_back(A); } 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<long long> dp1(N, 0), dp2(N, 0); for (int v : b){ int cnt = c[v].size(); vector<vector<long long>> dp(2, vector<long long>(cnt + 1, 0)); dp[0][0] = 1; dp[1][0] = 1; for (int i = 0; i < cnt; i++){ int w = c[v][i]; dp[0][i + 1] += dp[0][i] * (dp1[w] + dp2[w]) % MOD; dp[0][i + 1] %= MOD; dp[1][i + 1] += dp[0][i] * dp2[w] % MOD; dp[1][i + 1] += dp[1][i] * (dp1[w] + dp2[w]) % MOD; dp[1][i + 1] %= MOD; } dp1[v] = dp[0][cnt]; dp2[v] = dp[1][cnt]; } cout << dp2[0] << endl; }