CF1481E

$* \text{Defficult:} \color{Red}{2500}$

一道很有 AT 风格的 DP。

Description

有 $n$ 本书，每本书有一个颜色，每次操作可以将一本书移动到最右。

求问使所有相同颜色的书相邻的最小移动次数。

$n \le 5 \cdot 10^5$

Solution

首先可以想到，最大的移动次数不会超过 $n$ ，且每本书的移动次数不会超过一次。

因为显然，所有书按照颜色排序然后分别移动即可。

反向考虑，设法求出不用移动的书的最大值。

对于每种颜色的数，记录其最左边出现的位置 $l_i$ ，和最右边出现的位置 $r_i$ 。

设 $f_i$ 表示使位置在范围 $[i,n]$ 之内的书按照要求排列的最大不移动的书的数量。

考虑如何转移：

首先，所有情况下的 $i$ ，都可以选择移动，使之与其他同颜色的书相邻，即 $f_i =f_{i+1}$ 。

转移的过程中，记录此时每种书的出现次数 $cnt_i$ 。

剩下的转移要分情况讨论。

$i = l_{a_i}$

此时可以选择不移动所有颜色为 $a_i$ 的书，有转移 $f_{i} = cnt_{a_i} + f_{r_{a_i} +1}$ 。

$i \ne l_i$

跟上面一样，考虑颜色为 $a_i$ 的书不移动。

此时的转移是 $f_i = cnt_{a_i}$ 。

比如考虑 $[1,2,2,1,1,3]$ ，当前 $i=4$ 的情况。

如果选择上述转移，实际上的操作是：

$[1,2,2,1,1,3] \implies [2,2,1,1,3,1] \implies [2,2,1,1,1,3]$ ，操作次数为 $2$ ，等于 $cnt_{a_i}$ 。

总结一下，状态转移方程式是：

f_{i} = max {\begin{cases} f_{i + 1} \\ c n t_{a_{i}} & i \neq l_{a_{i}} \\ f_{r_{a_{i}} + 1} & i = l_{a_{i}} \end{cases}

$f_i = \max \begin{cases}f_{i+1}\\cnt_{a_i}&i \ne l_{a_i}\\f_{r_{a_i}+1}&i=l_{a_i}\end{cases}$

最后的答案就是 $n-f_1$ 。

时间复杂度： $O(n)$ 。

code

#include <bits/stdc++.h>
using namespace std;
#define L(i,j,k) for(int (i)=(j);i<=(k);(i)++)
#define R(i,j,k) for(int (i)=(j);i>=(k);(i)--)
#define FST ios::sync_with_stdio(0);cin.tie(0);
const int N=2e6+100;
int n,a[N],l[N],r[N],cnt[N],f[N];
signed main(){
  FST;cin>>n;L(i,1,n) cin>>a[i],r[a[i]]=i;
  L(i,1,n) if(!l[a[i]]) l[a[i]]=i;
  R(i,n,1){
    cnt[a[i]]++;if(i==l[a[i]]) 
        f[i]=max(cnt[a[i]]+f[r[a[i]]+1],f[i+1]);
    else f[i]=max(f[i+1],cnt[a[i]]);
  }cout<<n-f[1];
}//CSP-S 2022 RP++