JOI 2018/2019 予選 過去問から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
「競技プログラミング」を略して、「競プロ」などと呼ばれています。
#E - イルミネーション (Illumination)
僕が作成、提出したコードは、以下のとおりです。
/* AtCoder Problems in C++ #E - イルミネーション (Illumination) https://atcoder.jp/contests/joi2019yo/tasks/joi2019_yo_e 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; int main(void){ ll n,m; cin>>n>>m; vector<ll> a(n); for(ll i=0;i<n;i++){ cin>>a[i]; } vector<array<ll,2>> k(m); for(ll i=0;i<m;i++){ cin>>k[i][0]>>k[i][1]; k[i][0]--; } sort(k.begin(),k.end()); k.push_back({998244353,998244353}); vector<ll> left(n+1); for(ll i=1,cnt=0;i<=n;i++){ while(cnt<=m && k[cnt][1]<i){ cnt++; } left[i]=min(i-1,k[cnt][0]); } vector<array<ll,2>> dp(n+1); for(ll i=0;i<n;i++){ dp[i+1][0]=max(dp[i][0],dp[i][1]); dp[i+1][1]=max(dp[left[i+1]][0],dp[left[i+1]][1])+a[i]; } cout<<max(dp[n][0],dp[n][1])<<endl; return 0; }