[BZOJ]5125: [Lydsy1712月赛]小Q的书架 DP+决策分治

题解：

$f_{i,j}$表示前$i$个数分成$j$段的最小代价，显然对于同一个$j$，决策端点是单调右移的，所以可以用分治来优化转移。$solve(j,l,r,L,R)$表示求出$f_{l,j}$到$f_{r,j}$，这些状态的决策点为$[L,R]$。求出$mid$，暴力求解其决策点，设为$k$，然后递归$solve(j,l,mid-1,L,k)$，$solve(j,mid+1,r,k,R)$即可。求区间逆序对可以类似莫队维护。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=40010;
const int Maxk=12;
const int inf=2147483647;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int n,k,a[Maxn],nl=1,nr=1;LL f[Maxn][Maxk];
LL cost=0,s[Maxn];
void add(int x,LL y){if(!x)return;for(;x<=n;x+=(x&-x))s[x]+=y;}
LL query(int x){LL re=0;for(;x;x-=(x&-x))re+=s[x];return re;}
void move(int L,int R)
{
    while(nl<L)add(a[nl],-1),cost-=query(a[nl]-1),nl++;
    while(nr>R)add(a[nr],-1),cost-=(nr-nl-query(a[nr])),nr--;
    while(nl>L)nl--,add(a[nl],1),cost+=query(a[nl]-1);
    while(nr<R)nr++,add(a[nr],1),cost+=(nr-nl+1-query(a[nr]));
}
void solve(int p,int l,int r,int L,int R)//求解区间[l,r]  决策区间[L,R]
{
    if(l>r)return;
    int mid=l+r>>1,w;LL v=(1LL<<60);
    for(int i=min(R,mid-1);i>=L;i--)
    {
        move(i+1,mid);
        LL t=f[i][p-1]+cost;
        if(t<v)v=t,w=i;
    }
    f[mid][p]=v;
    solve(p,l,mid-1,L,w);solve(p,mid+1,r,w,R);
} 
int main()
{
    memset(f,63,sizeof(f));
    n=read(),k=read();
    for(int i=1;i<=n;i++)a[i]=read();
    f[0][0]=0;
    add(a[1],1);
    for(int K=1;K<=k;K++)solve(K,1,n,0,n-1);
    printf("%lld",f[n][k]);
}