[ 决策单调性优化DP ] 计蒜客 是男人就过 8 题 H Sign Location

%%%LargestJN

显然标志放在车站上是最优的。
令 $f_{i,j}$ 表示放了前 $i$ 个标志，最后一个标志在 $j$ 位置的最小代价，转移推下式子就行了。
打个暴力发现满足决策单调性，分治做就好了。
讲下怎么推式子。
最终 $k$ 个标志把 $n$ 个车站分成了 $k+1$ 个区间，我们把 $(i,j)$ 的贡献算在 $i$ 所在的区间内，然后统计每个区间的贡献。
假设当前要放的位置是 $r$ ，上一个位置是 $l$ ，那么 $i$ 在 $[l+1,r-1]$ 内。

• $l< j\le r$ ：
  

  $$2\sum_{k=l+2}^{r-1} \sum_{p=l+1}^{k-1}(x_k-x_p)=2\sum_{k=l+2}^{r-1} x_k(k-l-1)-2\sum_{k=l+1}^{r-2}x_k(r-k-1)$$

• $j\le l$ ：
  

  $$\sum_{k=l+1}^{r-1} (x_k-x_l)l=\sum_{k=l+1}^{r-1}x_kl-x_l(r-l-1)l$$

• $j>r$ ：
  $$\sum_{k=l+1}^{r-1}(x_r-x_k)(n-r+1)=x_r(r-l-1)(n-r+1)-\sum_{k=l+1}^{r-1}x_k(n-r+1)$$

预处理出 $x_i$ 的前缀和、$ix_i$ 的前缀和就好了。最后一段要特殊考虑。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=10010;
int k,n,m,cur;
int a[N];
ll p[N],g[N];
ll f[210][N],Ans;
ll Get(int i,int j) {
    ll Ans=2*(g[j-1]-g[i+1]+g[j-2]-g[i]-(p[j-1]-p[i+1])*(i+1)-(p[j-2]-p[i])*(j-1));
    Ans+=1ll*i*(p[j-1]-p[i]-1ll*a[i]*(j-i-1));
    Ans+=1ll*(n-j+1)*(1ll*a[j]*(j-i-1)-p[j-1]+p[i]);
    return Ans;
}
ll Get1(int i,int j) {
    if(i==j) return 0;
    ll Ans=2*(g[j]-g[i+1]+g[j-1]-g[i]-(p[j]-p[i+1])*(i+1)-(p[j-1]-p[i])*j);
    Ans+=1ll*i*(p[j]-p[i]-1ll*a[i]*(j-i));
    return Ans;
}
void Solve(int l,int r,int L,int R) {
    if(l>r) return;
    int Mid=l+r>>1;
    ll mn=2e18;int pos=L;
    for(int i=min(R,Mid-1);i>=L;i--) {
        ll t=f[cur-1][i]+Get(i,Mid);
        if(t<mn) mn=t,pos=i;
    }
    f[cur][Mid]=mn;
    Solve(l,Mid-1,L,pos);Solve(Mid+1,r,pos,R);
}
int main() {
    while(~scanf("%d%d",&n,&k)) {
        k=min(k,n);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]),p[i]=p[i-1]+a[i],g[i]=g[i-1]+1ll*a[i]*i;
        if(!k) {
            Ans=0;
            for(int i=2;i<=n;i++) Ans+=1ll*a[i]*i-p[i];
            printf("%lld\n",Ans<<1);
            continue;
        }
        for(int j=1;j<=k;j++) {
            cur=j;
            if(j==1) Solve(1,n,0,0);else
            Solve(j,n,j-1,n-1);
        }
        Ans=2e18;
        for(int i=k;i<=n;i++) Ans=min(Ans,f[k][i]+Get1(i,n));
        printf("%lld\n",Ans);
    }
    return 0;
}