結果

問題 No.2434 RAKUTAN de RAKUTAN
ユーザー Astral__
提出日時 2023-08-12 13:32:46
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 79 ms / 2,000 ms
コード長 1,700 bytes
コンパイル時間 2,235 ms
コンパイル使用メモリ 204,240 KB
最終ジャッジ日時 2025-02-16 02:59:25
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main() {
  int N, H, X;
  cin >> N >> H >> X;
  int G, B;
  cin >> G;
  set<int> g;
  set<int> b;
  for(int i = 1; i <= G; i++) {
    int a;
    cin >> a;
    g.insert(a);
  }
  cin >> B;
  for(int i = 1; i <= B; i++) {
    int a;
    cin >> a;
    b.insert(a);
  }
  
  H = min(H, B); //リープする回数はせいぜいB回

  set<int> remains;//圧縮する際、残るマスの座標を入れる　//簡単のため0マス目は圧縮に含めない

  for(int a : b) {
    int l = max(1, a - X);
    int r = min(N, a + X);
    for(int j = l; j <= r; j++) {
      remains.insert(j);
    }
  }

  int ans_base = 0; //確実に取れる単位を記録する変数

  for(int a : g) {
    if(!remains.count(a)) {
      ans_base++; //周辺に落単マスがないのならば確実に取れる
    }
  }
 
  int N1 = remains.size();//圧縮後に残ったマスの数 (0マス目は数に含めていない事に注意）

  vector<int> nw(N1 + 1, 0);//変換後のマスの情報
  int idx = 1;
  for(int a : remains) {
    if(b.count(a)) {
      nw[idx]--;
    }
    if(g.count(a)) {
      nw[idx]++;
    }
    idx++;
  }

  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 + ans_base << endl;
}
0