結果

問題 No.2434 RAKUTAN de RAKUTAN
コンテスト
ユーザー Astral__
提出日時 2023-08-12 12:07:06
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,653 bytes
コンパイル時間 2,127 ms
コンパイル使用メモリ 200,632 KB
最終ジャッジ日時 2025-02-16 02:47:52
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 7 RE * 5 MLE * 1 -- * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;

int main() {
  int N, H, X;
  cin >> N >> H >> X;
  int G, B;
  cin >> G;
  vector<int> old(N+1, 0);//変換前のマス。0なら授業、1なら得単、-1なら落単
  vector<int> g(G+1);
  for(int i = 1; i <= G; i++) {
    cin >> g[i];
    old[g[i]] = 1;
  }
  cin >> B;
  vector<int> b(B+1);
  for(int i = 1; i <= B; i++) {
    cin >> b[i];
    old[b[i]] = -1;
  }
  
  H = min(H, B); //リープする回数はせいぜいB回

  vector<int> count(N+1, 0);//iマス目の周辺に落単マスがあるならばcount[i]は1以上
  for(int i = 1; i <= N; i++) {
    if(old[i] == -1) {
      int l = max(0, i-X);
      int r = min(N, i+X);
      for(int j = l; j <= r; j++) {
        count[j]++;
      }
    }
  }

  vector<int> nw(1, 0);//変換後のマスの情報
  for(int i = 1; i <= N; i++) {
    if(count.at(i) != 0) {
      nw.push_back(old[i]);
    }
    else {
      int sum = 0;
      sum += old[i];
      while(i + 1 <= N && count[i+1] == 0) {
        i++;
        sum += old.at(i);
      }
      nw.push_back(sum);
    }
  }

  int N1 = nw.size() - 1;//変換後のマス目数
  vector<vector<int>> dp(N1+1, vector<int>(H+1, -100100100));
  dp.at(0).at(H) = 0;

  for(int i = 1; i <= N1; i++) {
    for(int j = 0; j <= H; j++) {
       if(j != H && i >= X) {
        dp[i][j] = max(dp[i][j], dp[i-X][j+1] + nw[i]);
       }
       dp[i][j] = max(dp[i][j], dp[i-1][j] + nw[i]);
    }
  }

  int ans = -100100100;//単位の数はマイナスを取りうることに注意してください。
  for (int j = 0; j <= H; j++) {
    ans = max(ans, dp[N1][j]);
  }

  cout << ans << endl;
}
0