题解:
f i , j f_{i,j} fi,j表示前 i i i个数分成 j j j段的最小代价,显然对于同一个 j j j,决策端点是单调右移的,所以可以用分治来优化转移。 s o l v e ( j , l , r , L , R ) solve(j,l,r,L,R) solve(j,l,r,L,R)表示求出 f l , j f_{l,j} fl,j到 f r , j f_{r,j} fr,j,这些状态的决策点为 [ L , R ] [L,R] [L,R]。求出 m i d mid mid,暴力求解其决策点,设为 k k k,然后递归 s o l v e ( j , l , m i d − 1 , L , k ) solve(j,l,mid-1,L,k) solve(j,l,mid−1,L,k), s o l v e ( j , m i d + 1 , r , k , R ) solve(j,mid+1,r,k,R) 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]);
}