UVA 12594 Naming Babies 斜率DP

Naming Babies UVA - 12594

给一个序列 $PN$，该序列是 $a-z$ 的排列，再给一个序列 $NM=\overline{C_0{C}_1\cdots C_{n-1}}$，对于该序列的第 $i$ 个字符 $C_i$，定义 $PO{S}_i$ 为字符 $C_i$ 在序列 $PN$ 中的位置（从 $0$ 开始），定义总体花费为：

$$\sum _{i=0}^{n-1}(i-PO{S}_i)\ast PO{S}_i$$

题目给了个例子：

然后把这个序列分成 $k$ 段，然后每段分别计算这个花费，之后再求和，问最小花费和是多少？

设 $dp[i][j]$ 为前 $i$ 个分 $j$ 段，则有递推方程：

d p [ i ] [ j ] = min ⁡ { d p [ k ] [ j − 1 ] + ∑ m = k + 1 i ( m − k − 1 − p [ a [ m ] ] ) ∗ p [ a [ m ] ] } d p [ i ] [ j ] = min ⁡ { d p [ k ] [ j − 1 ] + s [ i ] − s [ k ] − ( k + 1 ) ( s p [ i ] − s p [ k ] ) } \begin{aligned} dp[i][j]&=\min\{dp[k][j-1]+\sum_{m=k+1}^{i}(m-k-1-p[a[m]])*p[a[m]]\}\\ dp[i][j]&=\min\{dp[k][j-1]+s[i]-s[k]-(k+1)(sp[i]-sp[k])\} \end{aligned} dp[i][j]dp[i][j]​=min{dp[k][j−1]+m=k+1∑i​(m−k−1−p[a[m]])∗p[a[m]]}=min{dp[k][j−1]+s[i]−s[k]−(k+1)(sp[i]−sp[k])}​

其中 $s[i]={\sum }_{m=0}^i(m-p[a[m]])p[a[m]],sp[i]={\sum }_{m=0}^{i}p[a[m]]$，$a=NM,p[C_i]=PO{S}_i$ ，设 $k_1<k_2$，且 $k_2$ 优于 $k_1$，则：

$$dp[k_2][j-1]+s[i]-s[k_2]-(k_2+1)(sp[i]-sp[k_2])<dp[k_1][j-1]+s[i]-s[k_1]-(k_1+1)(sp[i]-sp[k_1])$$

$$\begin{array}{rl}dp[k_2][j-1]-s[k_2]+sp[k_2](k_2+1)-(dp[k_1][j-1]-s[k_1]+sp[k_1](k_1+1))& <sp[i](k_2-k_1)\\ \frac{y_{k_2}-y_{k_1}}{x_{k_2}-x_{k_1}}& <sp[i]\end{array}$$

其中 $y_k=dp[k][j-1]-s[k]+sp[k](k+1),x_k=k$。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
//#define WINE
#define MAXN 20010
using namespace std;
typedef long long ll;
int T,iCase,k,n,p[300],h,t,q[MAXN];
ll s[MAXN],sp[MAXN],dp[MAXN][505];
char PN[30],a[MAXN];
ll up(int k2,int k1,int j){
    return dp[k2][j-1]-s[k2]+(k2+1)*sp[k2]-(dp[k1][j-1]-s[k1]+(k1+1)*sp[k1]);
}
ll down(int k2,int k1){
    return k2-k1;
}
ll getDP(int k,int i,int j){
    return dp[k][j-1]+s[i]-s[k]-(k+1)*(sp[i]-sp[k]);
}
int main(){
#ifdef WINE
    freopen("data.in","r",stdin);
#endif
    scanf("%d",&T);
    while(T--){
        scanf(" %s %d",PN,&k);
        scanf(" %s ",a);
        int len=strlen(PN);
        for(int i=0;i<len;i++)p[PN[i]]=i;
        n=strlen(a);
        for(int i=0;i<n;i++){
            s[i]=(i==0)?0:s[i-1];
            s[i]+=(i-p[a[i]])*p[a[i]];
            sp[i]=(i==0)?0:sp[i-1];
            sp[i]+=p[a[i]];
            dp[i][1]=s[i];
        }
        for(int j=2;j<=k;j++){
            h=t=0;q[t++]=j-2;
            for(int i=j-1;i<n;i++){
                while(h+1<t&&up(q[h+1],q[h],j)<sp[i]*down(q[h+1],q[h]))
                    h++;
                dp[i][j]=getDP(q[h],i,j);
                while(h+1<t&&up(i,q[t-1],j)*down(q[t-1],q[t-2])<=up(q[t-1],q[t-2],j)*down(i,q[t-1]))
                    t--;
                q[t++]=i;
            }
        }
        printf("Case %d: %lld\n",++iCase,dp[n-1][k]);
    }
    return 0;
}