結果
問題 | No.2214 Products on Tree |
ユーザー |
![]() |
提出日時 | 2023-02-10 22:03:28 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 184 ms / 3,000 ms |
コード長 | 758 bytes |
コンパイル時間 | 3,417 ms |
コンパイル使用メモリ | 251,236 KB |
最終ジャッジ日時 | 2025-02-10 12:47:38 |
ジャッジサーバーID (参考情報) |
judge1 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint998244353; using namespace std; #define rep(i,n) for (int i = 0; i < (n); ++i) #define Inf32 1000000001 #define Inf64 1000000000000000001 mint dp[200005][2]; vector<vector<int>> E; void dfs(int c,int p){ dp[c][0] = 1; rep(i,E[c].size()){ int to = E[c][i]; if(to==p)continue; dfs(to,c); mint x = dp[to][0],y = dp[to][1]; dp[c][1] = dp[c][0] * y + dp[c][1] * (x + (y)); dp[c][0] = dp[c][0]*x + dp[c][0]*y; } dp[c][1] += dp[c][0]; } int main(){ int n; cin>>n; E.resize(n); rep(i,n-1){ int a,b; cin>>a>>b; a--,b--; E[a].push_back(b); E[b].push_back(a); } dfs(0,-1); cout<<dp[0][1].val()<<endl; return 0; }