oj.uz

답안 #224891

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
224891 2020-04-19T04:40:26 Z rama_pang 조교 (CEOI16_popeala) C++14
17 / 100
 2000 ms 5376 KB 
#include <bits/stdc++.h>
using namespace std;

const int MAXT = 20005;
const int MAXN = 55;

const int INF = 2e9 + 100;
const int MOD = 1e9 + 9;
const int KEY = 63;

int N, T, S;
int Points[MAXT];

int POWKEY[MAXT];
int INVKEY[MAXT];

int ResultsHash[MAXN][MAXT]; // Res[N] = 111...111

int Power(int n, int x) {
  if (x == 0) return 1;
  int res = Power(n, x / 2);
  res = 1ll * res * res % MOD;
  if (x & 1) res = 1ll * res * n % MOD;
  return res;
}

void Precompute() {
  POWKEY[0] = INVKEY[0] = 1;
  for (int i = 1; i < MAXT; i++) {
    POWKEY[i] = 1ll * POWKEY[i - 1] * KEY % MOD;
    INVKEY[i] = Power(POWKEY[i], MOD - 2);
  }
}

int GetHash(int id, int L, int R) {
  int res = (ResultsHash[id][R] - ResultsHash[id][L - 1]) % MOD;
  res = 1ll * res * INVKEY[L] % MOD;
  if (res < 0) res += MOD;
  return res;
}

int GetCost(int L, int R) {
  int cnt = 0;
  for (int i = 0; i < N; i++) {
    if (GetHash(i, L, R) == GetHash(N, L, R)) {
      cnt++;
    }
  }
  return cnt * (Points[R] - Points[L - 1]);
}

int dp[MAXN][MAXT];

void BaseCase() {
  for (int i = 0; i < MAXN; i++) {
    for (int j = 0; j < MAXT; j++) {
      dp[i][j] = INF;
    }
  }
  dp[0][0] = 0;
}

void Calc() {
  for (int i = 1; i <= T; i++) {
    for (int j = 1; j <= S; j++) {
      int &res = dp[j][i] = INF;
      for (int k = 0; k < i; k++) {
        res = min(res, dp[j - 1][k] + GetCost(k + 1, i));
      }
    }
  }
}

void DP_DnC(int s, int lo, int hi, int l, int r) {
  if (hi < lo) return;
  int mid = (lo + hi) / 2;
  int &res = dp[s][mid] = INF;
  int opt = -1;
  for (int k = l; k <= min(r, mid - 1); k++) {
    int cost = dp[s - 1][k] + GetCost(k + 1, mid);
    if (res > cost) {
      res = cost;
      opt = k;
    } 
  }
  DP_DnC(s, lo, mid - 1, l, opt);
  DP_DnC(s, mid + 1, hi, opt, r);
}

int main() {
  Precompute();

  cin >> N >> T >> S;
  for (int i = 1; i <= T; i++) {
    cin >> Points[i];
    Points[i] += Points[i - 1];
  }

  for (int i = 0; i <= N; i++) {
    string Result(T, '1');
    if (i < N) cin >> Result;
    auto Hash = ResultsHash[i];
    for (int j = 1; j <= T; j++) {
      Hash[j] = (1ll * Hash[j - 1] + 1ll * POWKEY[j] * (Result[j - 1] - '0' + 1)) % MOD;
    }
  }

  BaseCase();
  Calc();
  for (int i = 1; i <= S; i++) {
    // DP_DnC(i, 0, T, 0, T);
    cout << dp[i][T] << "\n";
  }

  return 0;
}

Subtask #1
8.0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 13 ms 4736 KB Output is correct
2 Correct 15 ms 4992 KB Output is correct

Subtask #2
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1755 ms 5240 KB Output is correct
2 Correct 1690 ms 5156 KB Output is correct
3 Correct 1799 ms 5164 KB Output is correct

Subtask #3
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 2075 ms 5376 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #4
0 / 74.0

# 결과 실행 시간 메모리 Grader output
1 Correct 13 ms 4736 KB Output is correct
2 Correct 15 ms 4992 KB Output is correct
3 Correct 1755 ms 5240 KB Output is correct
4 Correct 1690 ms 5156 KB Output is correct
5 Correct 1799 ms 5164 KB Output is correct
6 Execution timed out 2075 ms 5376 KB Time limit exceeded
7 Halted 0 ms 0 KB -