寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #E - イルミネーション (Illumination)

JOI 2018/2019 予選 過去問から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
競技プログラミング」を略して、「競プロ」などと呼ばれています。

#E - イルミネーション (Illumination)

atcoder.jp

僕が作成、提出したコードは、以下のとおりです。

/*
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;
}