CDOJ 879 摩天轮 dp+斜率优化

原题链接：http://www.acm.uestc.edu.cn/#/problem/show/879

题意：

中文题

题解：

这是一道斜率dp的题。

先把$a$数组排个序。

令$dp[i][j]$表示第$i$个人坐在第$j$个箱子里面的最优解。

容易得到以下转移方程：

$$dp[i][j]=min \left \{ dp[k][j-1]+(a[i]-a[k+1])^2 \right \}$$

观察这个式子，发现好像可以斜率优化：

若$u>v$且$u$要比$v$优，即：

$$dp[u][j-1]+(a[i]-a[u+1])^2<dp[v][j-1]+(a[i]-a[v+1])^2$$

整理后可得到下式：

$$\frac {dp[u][j-1]+a[u+1]^2-(dp[v][j-1]+a[v+1]^2)} {a[u+1]-a[v+1]} < 2*a[i]$$

令

$$Y(t)=dp[t][j-1]+a[t+1]^2$$

$$X(t)=a[t+1]$$

那么原式就是：

$$\frac {Y(u)-Y(v)} {X(u)-X(v)} <2*a[i]$$

这是一个描述斜率的东西，它表明，如果$u,v$的斜率小于$2*a[i]$，那么$u$一定比$v$更优，否则$v$不比$u$差。

令$S(u,v)=\frac {Y(u)-Y(v)} {X(u)-X(v)}$，那么如果$S(u,v)<S(v,w)$，那么$b$一定是没用的状态，这是因为假如$S(u,v)<2*a[i]$，那么$u$比$v$更优，如果$S(u,v) \geq 2*a[i]$，那么$S(v,w) \geq 2*a[i]$，那么$w$不比$v$差。所以$v$就是没用的节点。

综上，我们可以通过一个双端队列来维护一个下凸包，向队尾插入值，维护队尾的单调性。从队首寻找答案。

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#define MAX_N 100005
using namespace std;
 
int n,m;
int w[MAX_N];
 
int dp[MAX_N][];
 
double Y(int t,int j){
    return dp[t][j]+w[t+]*w[t+];
}
 
double X(int t){
    return w[t+];
}
 
double Slope(int u,int v,int j) {
    if (X(u) == X(v))return 1e233;
    return (Y(u, j) - Y(v, j)) / (X(u) - X(v));
}
 
int que[MAX_N];
 
int main() {
    scanf("%d%d", &n, &m);
    for (int i = ; i <= n; i++)scanf("%d", &w[i]);
    sort(w + , w +  + n);
    for (int i = ; i <= n; i++)dp[i][] = (w[i] - w[]) * (w[i] - w[]);
    for (int j = ; j <= m; j++) {
        int cur = j & ;
        int front = , rear = ;
        que[rear++] = ;
        for (int i = ; i <= n; i++) {
            while (rear - front >  && Slope(que[front], que[front + ], cur ^ ) <=  * w[i])
                front++;
            dp[i][cur] = dp[que[front]][cur ^ ] + (w[i] - w[que[front] + ]) * (w[i] - w[que[front] + ]);
            while (rear - front >  && Slope(que[rear - ], que[rear - ], cur ^ ) >= Slope(i, que[rear - ], cur ^ ))
                rear--;
            que[rear++] = i;
        }
    }
    printf("%d\n", dp[n][m & ]);
    return ;
}