B - Lawrence HDU - 2829 斜率dp dp转移方程不好写

B - Lawrence

 HDU - 2829 

这个题目我觉得很难,难在这个dp方程不会写。

看了网上的题解,看了很久才理解这个dp转移方程

dp[i][j] 表示前面1~j 位并且以 j 结尾分成了 i 段的最小权值和

再定义一个数组 w[a,b] 表示 a到b 的权值和,注意这个不是前缀和,而是题目给的那种权值和

比如 a 到 b  是4 5 1 2 Its Strategic Value is 4*5 + 4*1 + 4*2 + 5*1 + 5*2 + 1*2 = 49. 

w[a,b]=49

然后再定义一个val[i] ==w[1,i] 意思就是前缀权值和 val[i]=val[i-1]+sum[i-1]*a[i]

因为这个w[a,b]不好直接推出来,但是可以借助前缀来推。

求w[a+1,b]=val[b]-val[a]-(sum[b]-sum[a])*sum[a]

然后这个dp方程就可以推出来了

dp[i,j]=min(dp[i-1,k]+w[k+1,j])

但是这个直接暴力n^3肯定是过不了的,所以需要优化,这个式子是不是和之前推过的斜率优化的式子很像。

如果在dp[i-1,k]之外还存在k这个值,一般都是可以用斜率优化来优化这个dp的。

然后就可以推出和之前一样的式子(之前指的是D - Pearls HDU - 1300 斜率dp+二分  斜率dp A - Print Article HDU - 3507

令 F[h]=dp[i-1][h]-val[h]+sum[h]*sum[h]

所以G[h,k]=(F[h]-F[k)/(sum[h]-sum[k])<sum[j]

然后就是一样的推导了

如果存在 i>j>k G[i,j]>G[j,k]

1 G[i,j]>Gj,k]>sum[t]  k 比 j 优 ,j 比 i 优

2 G[i,j]>sum[t]>G[j,k] 那么 j 比 i 优,j 比 k 优

3 sum[t]>G[i,j]>G[j,k] i 比 j 优 j 比k 优

如果是 i > j > k G[i,j]<G[j,k]

1 G[i,j]<G[j,k]<sum[t] j 比 k 优 i 比 j 优

2. G[i,j]<sum[t]<G[j,k] i 比 j 优 k 比 j 优

3 sum[t]<G[i,j]<G[j,k]  j 比 i 优 k 比 j 优

这种情况之下,j 肯定是要被排除的,所以如果从后面插入 i 的时候 前面的 j 如果和它构成的斜率小于这个数和之前的那个数构成的斜率,

那么这个 j 肯定是不要的。

因为这个前缀和sum是单调的,所以可以用单调队列优化这个dp。

   

  

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <vector>
#include <iostream>
#include <string>
#define inf 0x3f3f3f3f
#define inf64 0x3f3f3f3f3f3f3f3f
using namespace std;
const int maxn = 1e5 + 10;
typedef long long ll;
ll dp[1100][1100], sum[1100], val[1100], a[1100];
int que[maxn];

ll up(int i,int j,int k)
{
    return dp[i - 1][j] - val[j] + sum[j] * sum[j] - (dp[i - 1][k] - val[k] + sum[k] * sum[k]);
}

ll down(int j,int k)
{
    return sum[j] - sum[k];
}

ll DP(int i,int j,int k)
{
    return dp[i - 1][k] + val[j] - val[k] - (sum[j] - sum[k])*sum[k];
}

int main()
{
    int n, m;
    while (scanf("%d%d", &n, &m) && (n + m)) {
        sum[0] = 0;
        val[0] = 0;
        for (int i = 1; i <= n; i++) {
            scanf("%lld", &a[i]);
            sum[i] = sum[i - 1] + a[i];
            val[i] = val[i - 1] + sum[i - 1] * a[i];
        }
        for (int i = 1; i <= n; i++) dp[1][i] = val[i];//注意这个题目的初始化
        for (int i = 2; i <= m + 1; i++) {
            int head = 0, tail = 0;//注意dp[i,j]的定义是以j结尾分成了i块的最小价值
            que[tail++] = i - 1;//因为下一段要分成i块,所以前面最少也占了i-1个数,而后面又有可能全部分成一块,注意这种情况不要漏掉
            for (int j = i; j <= n; j++) {//注意这个j必须要从i开始
                while (head + 1 < tail&&up(i, que[head + 1], que[head]) <= sum[j] * down(que[head + 1], que[head])) head++;
                dp[i][j] = DP(i, j, que[head]);
                while (head + 1 < tail&&up(i, j, que[tail - 1])*down(que[tail - 1], que[tail - 2]) <= up(i, que[tail - 1], que[tail - 2])*down(j, que[tail - 1])) tail--;
                que[tail++] = j;
            }
        }
        printf("%lld\n", dp[m + 1][n]);
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/EchoZQN/p/11402172.html