結果

問題 No.2047 Path Factory
ユーザー kotatsugamekotatsugame
提出日時 2022-08-19 21:54:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 112 ms / 2,000 ms
コード長 623 bytes
コンパイル時間 779 ms
コンパイル使用メモリ 72,596 KB
実行使用メモリ 20,608 KB
最終ジャッジ日時 2024-10-15 13:30:11
合計ジャッジ時間 3,001 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<atcoder/modint>
using namespace std;
using mint=atcoder::modint998244353;
int N;
vector<int>G[1<<17];
mint dp[1<<17][3];
void dfs(int u,int p)
{
	mint a=1,b=0,c=0;
	for(int v:G[u])if(v!=p)
	{
		dfs(v,u);
		mint na=a*(dp[v][0]+dp[v][1]);
		mint nb=b*(dp[v][0]+dp[v][1])+a*(dp[v][1]+dp[v][2]);
		mint nc=c*(dp[v][0]+dp[v][1])+b*(dp[v][1]+dp[v][2]);
		a=na,b=nb,c=nc;
	}
	dp[u][0]=c;
	dp[u][1]=b;
	dp[u][2]=a;
}
int main()
{
	cin>>N;
	for(int i=1;i<N;i++)
	{
		int 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]).val()<<endl;
}
0