Submission #1087539

#	Time	Username	Problem	Language	Result	Execution time	Memory
1087539		juicy	Popeala (CEOI16_popeala)	C++17	100 / 100	221 ms	9956 KiB

popeala

#include <bits/stdc++.h>
 
using namespace std;
 
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) 42
#endif
 
const int N = 20005, D = 55, inf = 2e9;
 
int n, t, s;
int a[N], mi[D], lt[N][D], dp[D][N];
 
int main() {
  ios::sync_with_stdio(false); cin.tie(nullptr);
 
  cin >> n >> t >> s;
  for (int i = 1; i <= t; ++i) {
    cin >> a[i];
    a[i] += a[i - 1];
  }
  for (int i = 0; i < n; ++i) {
    string s; cin >> s;
    for (int j = 0; j < t; ++j) {
      lt[j + 1][i] = s[j] == '1' ? lt[j][i] : j + 1;
    }
  }
  for (int i = 1; i <= t; ++i) {
    lt[i][n] = i;
    sort(lt[i], lt[i] + n + 1);
  }
  fill(dp[0] + 1, dp[0] + t + 1, inf);
  for (int i = 1; i <= s; ++i) {
    fill(mi, mi + n + 1, inf);
    dp[i][0] = inf;
    for (int j = 1; j <= t; ++j) {
      dp[i][j] = inf;
      for (int k = 0; k <= n; ++k) {
        for (int p = lt[j - 1][k]; p < lt[j][k]; ++p) {
          if (dp[i - 1][p] ^ inf) {
            mi[k] = min(mi[k], dp[i - 1][p] - a[p] * k);
          }
        }
        if (mi[k] ^ inf) {
          dp[i][j] = min(dp[i][j], mi[k] + a[j] * k);
        }
      }
    }
    cout << dp[i][t] << "\n";
  }
  return 0;
}

• Subtask 1: 8 / 8
• Subtask 2: 9 / 9
• Subtask 3: 9 / 9
• Subtask 4: 74 / 74