首页 > 代码库 > HDU2829 Lawrence(斜率优化dp)
HDU2829 Lawrence(斜率优化dp)
学了模板题之后上网搜下斜率优化dp的题目,然后就看到这道题,知道是斜率dp之后有思路就可以自己做不出来,要是不事先知道的话那就说不定了。
题意:给你n个数,一开始n个数相邻的数之间是被东西连着的,对于连着的一片的数,它们的价值就是两两乘积的和。所以4 5 1 2一开始就是4*5+4*1+4*2+5*1+5*2+1*2... 注意到两两乘积的和其实是可以这么算的((a1+a2+a3+..an)^2-(a1^2+a2^2+....))/2。现在我可以在数与数之间切m刀,问切完之后的最小价值是多少。
一个自然的想法是定义dp[n][t]表示的是前n个数被切了t刀的最小价值,不难发现方程是可以这么转移的
dp[i][t]=dp[j][t-1]+((sum[i]-sum[j])*(sum[i]-sum[j])-(d[i]-d[j]))/2 (j<i) 其中d[i]表示前i个数的平方的和
所以这个方程是1个三维的dp,对于1000的数据量是行不通的,所以仿照斜率优化的思路,我们可以尝试写出当 k<j时,j比k更优的方程,即:
dp[j][t-1]+((sum[i]-sum[j])*(sum[i]-sum[j])-(d[i]-d[j]))/2 < dp[k][t-1]+((sum[i]-sum[k])*(sum[i]-sum[k])-(d[i]-d[k]))/2
经过化简,我们可以得到一个这样的东西:
(2dp[j]+sum[j]^2+d[j])-(2dp[k]+sum[k]^2+d[k])/2*(sum[j]-sum[k]) < sum[i]
显然又是一个斜率式子,利用先前推导的性质我们可以知道,我们每次更新的时候其实就是利用t-1层的信息推出t层的信息,推的时候实际有效的值也是一个下凸的点集,这个和上一题不一样之处就在于要更新m次,每次队列都要重新入队,但总体来说还是一样的,由于斜率优化后里层的dp达到线性,所以最后复杂度出来是O(n^2)的,仍然要注意之所以一开始可以弹队首是因为 sum[i]是递增的。
#pragma warning(disable:4996)#include <iostream>#include <cstdio>#include <vector>#include <algorithm>#include <cstring>#include <string>#include <cmath>using namespace std;#define ll long long#define maxn 1100ll dp[maxn][maxn];ll a[maxn];ll sum[maxn];ll d[maxn];int n, m;ll getup(int i, int j, int t){ return (2 * dp[i][t] + sum[i] * sum[i] + d[i]) - (2 * dp[j][t] + sum[j] * sum[j] + d[j]);}ll getdown(int i, int j){ return 2 * (sum[i] - sum[j]);}int que[maxn];int qh, qt;int main(){ while (cin >> n >> m&&(n||m)){ a[0] = d[0] = sum[0] = 0; for (int i = 1; i <= n; ++i){ scanf("%I64d", &a[i]); sum[i] = sum[i - 1] + a[i]; d[i] = d[i - 1] + a[i] * a[i]; } dp[0][0] = 0; for (int i = 1; i <= n; ++i){ dp[i][0] = dp[i - 1][0] + a[i] * sum[i - 1]; } for (int x = 1; x <= m; ++x){ dp[0][x] = 0; qh = qt = 0; que[qt++] = 0; for (int i = 1; i <= n; ++i){ while (qh + 1 < qt && getup(que[qh + 1], que[qh], x - 1) <= sum[i] * getdown(que[qh + 1], que[qh])){ qh++; } dp[i][x] = dp[que[qh]][x - 1] + ((sum[i] - sum[que[qh]])*(sum[i] - sum[que[qh]]) - (d[i] - d[que[qh]])) / 2; while (qh + 1 < qt && getup(i, que[qt - 1], x - 1)*getdown(que[qt - 1], que[qt - 2]) <= getup(que[qt - 1], que[qt - 2], x - 1)*getdown(i, que[qt - 1])){ qt--; } que[qt++] = i; } } printf("%I64d\n", dp[n][m]); } return 0;}