-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathatcoder_dp.cpp
More file actions
46 lines (44 loc) · 803 Bytes
/
atcoder_dp.cpp
File metadata and controls
46 lines (44 loc) · 803 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll dp[100000][2];
ll mod = 1e9+7;
ll dfs(vector<vector<int>>&graph, int sou, int par, int colour ){
if(dp[sou][colour]!=-1){
return dp[sou][colour];
}
ll res = 1;
for(auto it:graph[sou]){
if(it!=par){
if(colour==1){
res *= dfs(graph,it,sou,0) + dfs(graph,it,sou,1);
res=res%mod;
}
else{
res *= dfs(graph,it,sou,1);
res=res%mod;
}
}
}
return dp[sou][colour]=res;
}
int main(){
int x;
cin>>x;
memset(dp,-1,sizeof(dp));
vector<vector<int>>graph(x);
for(int i=0;i<x-1;i++){
int sou,des;
cin>>sou>>des;
sou--;
des--;
graph[sou].push_back(des);
graph[des].push_back(sou);
}
dfs(graph,0,-1,0);
dfs(graph,0,-1,1);
ll ans = dp[0][0] + dp[0][1];
ans=ans%mod;
cout<<ans;
return 0;
}