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