Submission #235505

#	Time	Username	Problem	Language	Result	Execution time	Memory
235505		doowey	Holding (COCI20_holding)	C++14	110 / 110	992 ms	45176 KiB

holding

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;

#define fi first
#define se second
#define mp make_pair
#define fastIO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);

const int N = 102;
const int S = (int)2e4 + 100;
const int M = S/2;
const int inf = (int)1e9;

int dpL[2][N][S]; // pick i items, indice sum j
int dpR[2][N][S];
int a[N];
int li[N][S];
int ri[N][S];

void upd(int &a, int b){  
  a = min(a, b);
}

int main(){
  fastIO;
  int n, l, r, k;
  cin >> n >> l >> r >> k;
  for(int q = 0; q < 2; q ++ ){
    for(int d = 0; d <= n; d ++ ){
      for(int s = 0; s < S; s ++ ){
        dpL[q][d][s] = inf;
        dpR[q][d][s] = inf;
      }
    }
  }
  dpL[0][0][M]=0;
  int ans = 0;
  for(int i = 1; i <= n; i ++ ){
    cin >> a[i];
    if(i >= l && i <= r) ans += a[i];
  }
  for(int i = 1; i <= r; i ++ ){
    for(int j = 0 ; j <= n ; j ++ ){
      for(int x = 0; x < S ; x ++ ){
        if(i < l){
          if(dpL[0][j][x] == inf) continue;
          dpL[1][j + 1][x - i] = min(dpL[1][j + 1][x - i], dpL[0][j][x] + a[i]);
          dpL[1][j][x] = min(dpL[1][j][x], dpL[0][j][x]);
        }
        else{
          if(dpL[0][j][x] == inf) continue;
          if(j) dpL[1][j - 1][x + i] = min(dpL[1][j - 1][x + i], dpL[0][j][x]);
          dpL[1][j][x] = min(dpL[1][j][x], dpL[0][j][x] + a[i]);
        }
      }
    }
    for(int j = 0 ; j <= n; j ++ ){
      for(int x = 0 ; x < S; x ++ ){
        dpL[0][j][x]=dpL[1][j][x];
        dpL[1][j][x]=inf;
      }
    }
    for(int x = 0; x < S; x ++ ){
      li[i][x]=dpL[0][0][x];
    }
  }
  for(int x = 0 ;x <= k; x ++ ){
    if(li[r][M + x] == 0){
      ans = min(ans, li[r][M + x]);
    }
  }
  for(int x = 0; x <= k ; x ++ ){
    ans = min(ans, li[r][x + M]);
  }
  dpR[0][0][M]=0;
  for(int i = n; i >= l ; i -- ){
    for(int j = 0 ; j <= n; j ++ ){
      for(int x = 0; x < S; x ++ ){
        if(dpR[0][j][x] == inf) continue;
        if(i > r){
          dpR[1][j + 1][x + i] =min(dpR[1][j + 1][x + i], dpR[0][j][x] + a[i]);
          dpR[1][j][x] = min(dpR[1][j][x], dpR[0][j][x]);
        }
        else{
          if(j) dpR[1][j - 1][x - i] = min(dpR[1][j - 1][x - i], dpR[0][j][x]);
          dpR[1][j][x] = min(dpR[1][j][x], dpR[0][j][x] + a[i]);
        }
      }
    }
    for(int j = 0 ; j <= n; j ++ ){
      for(int x = 0; x < S; x ++ ){
        dpR[0][j][x]=dpR[1][j][x];
        dpR[1][j][x]=inf;
      }
    }
    for(int x = 0 ; x < S; x ++ ){
      ri[i][x]=dpR[0][0][x];
    }
  }
  for(int x = 0; x <= k ; x ++ ){
    ans = min(ans, ri[l][x + M]);
  }
  for(int i = l ; i + 1 <= r; i ++ ){
    for(int x = 0; x <= k; x ++ ){
      if(li[i][x + M] == inf) continue;
      for(int y = 0; x + y <= k; y ++ ){
        if(ri[i + 1][y + M] == inf) continue;
        ans = min(ans, li[i][x + M] + ri[i + 1][y + M]);
      }
    }
  }
  cout << ans << "\n";
  return 0;
}

• Subtask 1: 22 / 22
• Subtask 2: 33 / 33
• Subtask 3: 33 / 33
• Subtask 4: 22 / 22