Submission #807532

#	Time	Username	Problem	Language	Result	Execution time	Memory
807532		dong_liu	Painting Walls (APIO20_paint)	C++17	100 / 100	616 ms	17584 KiB

paint

#include <iostream>
#include <set>
#include <vector>

#include "paint.h"
using namespace std;

typedef vector<int> vi;

int minimumInstructions(int n, int m, int k, vi a, vi c, vector<vi> b) {
  vector<vi> pos(k);
  for (int i = 0; i < m; i++)
    for (int x : b[i]) {
      pos[x].push_back(i);
    }
  vi cnt(m), start(n);
  for (int i = 0; i < n; i++) {
    vector<pair<int, int>> v;
    for (int j : pos[a[i]]) {
      int j_ = (j - 1 + m) % m;
      v.push_back({j, cnt[j_] + 1});
    }
    if (i > 0) {
      for (int j : pos[a[i - 1]]) cnt[j] = 0;
    }
    for (auto [j, x] : v) {
      if (x >= m) start[i - m + 1] = 1;
      cnt[j] = x;
    }
  }
  vi dp(n + 1, n + 1);
  dp[0] = 0;
  multiset<int> v;
  for (int i = 0; i <= n; i++) {
    if (v.size()) dp[i] = *v.begin() + 1;
    if (start[i]) {
      v.insert(dp[i]);
    }
    if (i >= m && start[i - m]) v.erase(v.find(dp[i - m]));
  }
  if (dp[n] > n) return -1;
  return dp[n];
}

• Subtask 1: 12 / 12
• Subtask 2: 15 / 15
• Subtask 3: 13 / 13
• Subtask 4: 23 / 23
• Subtask 5: 37 / 37