AtCoder Beginner Contestの過去問から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
「競技プログラミング」を略して、「競プロ」などと呼ばれています。
#D - 塗り絵
僕が作成、提出したコードは、以下のとおりです。
/* AtCoder Problems in C++ #D - 塗り絵 https://atcoder.jp/contests/abc036/tasks/abc036_d 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; const ll mod=1000000007; vector<vector<ll>> g; ll dp[100005][2]; void DFS(ll v,ll p){ dp[v][0]=1; dp[v][1]=1; for(auto i : g[v]){ if(i==p){ continue; } DFS(i,v); dp[v][0]=((dp[v][0]%mod)*((dp[i][0]+dp[i][1])%mod))%mod; dp[v][1]=((dp[v][1]%mod)*dp[i][0])%mod; } } int main(void){ ll n; cin>>n; g.resize(n); for(ll i=0;i<n-1;i++){ ll a,b; cin>>a>>b; a--; b--; g[a].push_back(b); g[b].push_back(a); } DFS(0,0); cout<<(dp[0][0]%mod+dp[0][1]%mod)%mod<<endl; return 0; }
AtCoder Beginner Contestは、オンラインジャッジによるプログラミングコンテストです。
日本語と英語に対応していて、週末ごとに実施されているみたいです。
https://practice.contest.atcoder.jp/tutorial
アカウントを登録すれば、誰でも参加できます。