CF 1527 E

題目描述

我們定義一個陣列 \(P\) 的代價為：

\[\sum \limits_{x\in P} last(x)-first(x) \]

這裡 \(first(x),last(x)\) 是指 \(x\) 第一次，最後一次出現的位置。

你需要將陣列 \(A\) 分成恰好 \(k\) 段，求最小總代價。

思路

令 \(dp_{i,j}\) 表示已經分了 \(i\) 段，末尾在 \(j\)​ 的最小代價。

由於代價是最後一個減前一個，可以看作是 \(i_2-i_1+i_3-i_2+\dots\)，這裡 \(i_j\) 表示第 \(j\) 個 \(x\) 的下標。

所以轉移可以這樣來實現：當列舉到 \(i\) 時，之後 \(a\rightarrow b(1\le a<last,i\le b\le N)\) 的轉移一定會多 \(i-last\) 的代價。這個可以用線段樹維護區間最小值。

每次做完一輪 dp 就把 \(dp\) 值丟進線段樹即可。

空間複雜度 \(O(N)\)，時間複雜度 \(O(NK\log N)\)。

程式碼

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

const int MAXN = 35005;

struct Segment_Tree {
  int l[MAXN << 2], r[MAXN << 2], Min[MAXN << 2], lazy[MAXN << 2];
  void build(int u, int s, int t) {
    l[u] = s, r[u] = t, lazy[u] = 0;
    if(s == t) {
      Min[u] = 0;
      return;
    }
    int mid = (s + t) >> 1;
    build(u << 1, s, mid), build((u << 1) | 1, mid + 1, t);
    Min[u] = min(Min[u << 1], Min[(u << 1) | 1]);
  }
  void tag(int u, int x) {
    Min[u] += x, lazy[u] += x;
  }
  void pushdown(int u) {
    tag(u << 1, lazy[u]), tag((u << 1) | 1, lazy[u]), lazy[u] = 0;
  }
  void update(int u, int s, int t, int x) {
    if(s > t) {
      return;
    }
    if(l[u] >= s && r[u] <= t) {
      tag(u, x);
      return;
    }
    pushdown(u);
    if(s <= r[u << 1]) {
      update(u << 1, s, t, x);
    }
    if(t >= l[(u << 1) | 1]) {
      update((u << 1) | 1, s, t, x);
    }
    Min[u] = min(Min[u << 1], Min[(u << 1) | 1]);
  }
  int Getmin(int u, int s, int t) {
    if(s > t) {
      return 0;
    }
    if(l[u] >= s && r[u] <= t) {
      return Min[u];
    }
    pushdown(u);
    int x = int(2e9);
    if(s <= r[u << 1]) {
      x = min(x, Getmin(u << 1, s, t));
    }
    if(t >= l[(u << 1) | 1]) {
      x = min(x, Getmin((u << 1) | 1, s, t));
    }
    return x;
  }
}tr;

int n, k, a[MAXN], last[MAXN], dp[MAXN];

int main() {
  ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  cin >> n >> k;
  for(int i = 2; i <= n + 1; ++i) {
    cin >> a[i];
  }
  tr.build(1, 1, n + 1);
  tr.update(1, 2, n + 1, int(2e9));
  for(int i = 1; i <= k; ++i) {
    for(int j = 1; j <= n; ++j) {
      last[j] = 1;
    }
    for(int j = 2; j <= n + 1; ++j) {
      tr.update(1, 1, last[a[j]] - 1, j - last[a[j]]);
      last[a[j]] = j;
      dp[j] = tr.Getmin(1, 1, j - 1);
    }
    tr.build(1, 1, n + 1);
    tr.update(1, 1, 1, int(2e9));
    for(int j = 2; j <= n + 1; ++j) {
      tr.update(1, j, j, dp[j]);
    }
  }
  cout << dp[n + 1];
  return 0;
}