結果

問題 No.2047 Path Factory
コンテスト
ユーザー CleverLarry0702
提出日時 2022-12-01 13:54:24
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 87 ms / 2,000 ms
コード長 606 bytes
コンパイル時間 1,833 ms
コンパイル使用メモリ 169,224 KB
実行使用メモリ 22,912 KB
最終ジャッジ日時 2024-10-08 14:46:08
合計ジャッジ時間 4,081 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define ll long long

const ll mod=998244353;
ll N;
vector<ll>G[1<<17];
ll dp[1<<17][3];

void dfs(ll u,ll p)
{
	ll a=1,b=0,c=0;
	for(ll v:G[u])if(v!=p)
	{
		dfs(v,u);
		ll na=a*(dp[v][0]+dp[v][1])%mod;
        ll nb=( b*(dp[v][0]+dp[v][1])+a*(dp[v][1]+dp[v][2]) )%mod;
		ll nc=( c*(dp[v][0]+dp[v][1])+b*(dp[v][1]+dp[v][2]) )%mod;
		a=na,b=nb,c=nc;
	}
	dp[u][0]=c;
	dp[u][1]=b;
	dp[u][2]=a;
}

int main()
{
	cin>>N;
	for(ll i=1;i<N;i++)
	{
		ll u,v;
		cin>>u>>v;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs(1,0);
	cout<<(dp[1][0]+dp[1][1])%mod <<endl;
}
0