Codeforces 868F. Yet Another Minimization Problem

Posted 2020-12-04 yuzao

Description

给出一个长度为 (n) 的序列,你需要将它分为 (k) 段,使得每一段的价值和最小,每一段的价值是这一段内相同的数的个数
题面

Solution

容易想到设 (f[i][j]) 表示前 (i) 个数分成 (j) 段的最小代价
(f[i][j]=min(f[k][j-1]+w(k+1,i)))
这个(DP)有决策单调性,可以分治优化
设 (solve(l,r,L,R)) 表示用区间 ([L,R]) 内的决策去更新 ([l,r]) 的函数,找到 (mid) 的决策点然后分治下去就可以了

对于 (w) 函数的处理还需要注意:
我们维护两个全局指针,每一次移动到指定的区间,这样就可以 (O(1)) 转移了,均摊指针的移动次数是 (O(n*log)) 的

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int n,a[N],m,t[N],p=1,q=0;ll f[N][22],tot=0;
inline void upd(int l,int r){
    while(q<r)q++,tot+=t[a[q]],t[a[q]]++;
    while(p>l)p--,tot+=t[a[p]],t[a[p]]++;
    while(q>r)t[a[q]]--,tot-=t[a[q]],q--;
    while(p<l)t[a[p]]--,tot-=t[a[p]],p++;
}
inline void solve(int l,int r,int L,int R,int k){
    if(l==r){
        for(int i=min(R,r);i>=L;i--)
            upd(i+1,l),f[l][k]=min(f[l][k],f[i][k-1]+tot);
        return ;
    }
    int mid=(l+r)>>1,st=0;
    for(int i=min(mid-1,R);i>=L;i--){
        upd(i+1,mid);
        if(tot+f[i][k-1]<f[mid][k])f[mid][k]=tot+f[i][k-1],st=i;
    }
    if(st)solve(l,mid,L,st,k),solve(mid+1,r,st,R,k);
    else solve(mid+1,r,L,R,k);
}
int main(){
  freopen("pp.in","r",stdin);
  freopen("pp.out","w",stdout);
  scanf("%d%d",&n,&m);
  for(int i=1;i<=n;i++)scanf("%d",&a[i]);
  memset(f,127/3,sizeof(f));f[0][1]=0;
  for(int i=1;i<=n;i++)f[i][1]=f[i-1][1]+t[a[i]],t[a[i]]++;
  memset(t,0,sizeof(t));
  for(int i=2;i<=m;i++)
      solve(1,n,1,n,i);
  cout<<f[n][m]<<endl;
  return 0;
}