CF 1839 D

题目描述

给定 \(N\) 个不同颜色的小球。你可以进行以下操作：

• 插入一个颜色为 \(0\) 的小球，此操作最多执行 \(k\) 次。
• 选择一个非零球，使得该球与至少一个 \(0\) 小球相邻。并把该小球移动到任意位置。这样会花费 \(1\) 的代价。

对于每个 \(1\le k \le N\) 求出将序列变成一个去除 \(0\) 后升序的序列所需最小代价。

思路

定义 \(dp_{i,j}\) 表示把 \([1,i]\) 排序，使用 \(j\) 个 \(0\) 小球所需的最小代价。

如果 \(a_i > a_{i-1}\)，那么我们有转移 \(dp_{i,j}\leftarrow dp_{i-1,j}\)。

否则，对于所有 \(0\le k < i-1且a_k<a_i\)，那么我们有 \(dp_{i,j}\leftarrow dp_{k,j-1}+i-k-1\)，因为此时使用了一个 \(0\) 球，并且除端点外区间中每个点都需要移动一次。

空间复杂度 \(O(N^2)\)，时间复杂度 \(O(N^3)\)。

代码

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

const int MAXN = 505, INF = MAXN;

int t, n, a[MAXN], dp[MAXN][MAXN], ans;

void Solve() {
  cin >> n;
  for(int i = 1; i <= n; ++i) {
    cin >> a[i];
  }
  for(int i = 0; i <= n + 1; ++i) {
    for(int j = 0; j <= n; ++j) {
      dp[i][j] = INF;
    }
  }
  dp[0][0] = 0, a[n + 1] = n + 1;
  for(int i = 1; i <= n + 1; ++i) {
    for(int j = 0; j <= n; ++j) {
      if(a[i] > a[i - 1]) {
        dp[i][j] = dp[i - 1][j];
      }
      if(j) {
        for(int k = 0; k < i - 1; ++k) {
          if(a[k] < a[i]) {
            dp[i][j] = min(dp[i][j], dp[k][j - 1] + i - k - 1);
          }
        }
      }
    }
  }
  ans = dp[n + 1][0];
  for(int i = 1; i <= n; ++i) {
    ans = min(ans, dp[n + 1][i]);
    cout << ans << " \n"[i == n];
  }
}

int main() {
  ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  for(cin >> t; t--; Solve()) {
  }
  return 0;
}