結果

問題 No.2047 Path Factory
コンテスト
ユーザー okkuukenken
提出日時 2022-08-19 23:19:53
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,178 bytes
コンパイル時間 1,784 ms
コンパイル使用メモリ 155,496 KB
実行使用メモリ 20,736 KB
最終ジャッジ日時 2024-10-08 10:50:41
合計ジャッジ時間 4,676 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other AC * 11 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include<iostream>
#include<vector>
#include<algorithm>
#include<cmath>
#include<string>
#include<iomanip>
#include<numeric>
#include<queue>
#include<deque>
#include<stack>
#include<set>
#include<map>
#include<random>
#include<bitset>
#include<cassert>
#include<complex>
#include<fstream>
#include<cstdlib>
#include<functional>
#include<array>
using namespace std;
typedef long long ll;
const int mod=998244353;
const int dx[]={1,0,0,-1},dy[]={0,1,-1,0};
vector<int>G[100000];
int dp[2][100000];
void dfs(int v,int p){
	vector<array<int,2>>vec;
	for(auto&x:G[v]){
		if(x==p)
			continue;
		dfs(x,v);
		vec.push_back(array<int,2>({dp[0][x],dp[1][x]}));
	}
	vector<array<int,3>>dp2(vec.size()+1);
	dp2[0]={1,0,0};
	for(int i=0;i<vec.size();i++){
		for(int j=0;j<3;j++){
			dp2[i+1][j]=(dp2[i+1][j]+(ll)dp2[i][j]*vec[i][0])%mod;
			if(j!=2)
				dp2[i+1][j+1]=(dp2[i+1][j]+(ll)dp2[i][j]*vec[i][1])%mod;
		}
	}
	dp[0][v]=(dp2[vec.size()][1]+dp2[vec.size()][2])%mod;
	dp[1][v]=(dp2[vec.size()][0]+dp[0][v])%mod;
}
int main(){
	int n,u,v;
	cin>>n;
	while(cin>>u>>v){
		u--,v--;
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs(0,-1);
	cout<<dp[0][0]<<endl;
}
0