poj 3709 K-Anonymous Sequence

http://poj.org/problem?id=3709

/*
O(n^2)的DP方程：f[i]=Min{f[j]+sum[i]-sum[j]-a[j+1]*(i-j)}。

假设决策j1<j2并且j2优于（或者不差于）j1，那么

f[j1]+sum[i]-sum[j1]+a[j1+1]*(i-j1) >= f[j2]+sum[i]-sum[j2]-a[j2+1]*(i-j2)

[(f[j1]-sum[j1]+a[j1+1]*j1) - (f[j2]-sum[j2]+a[j2+1]*j2)] >= i*(a[j1+1]-a[j2+1])。

a[j2+1]>=a[j1+1]， 所以a[j1+1]-a[j2+1] <= 0。

可以写成：[(f[j1]-sum[j1]+a[j1+1]*j1) - (f[j2]-sum[j2]+a[j2+1]*j2)] / (a[j1+1]-a[j2+1]) <= i。

对于a[j1+1]==a[j2+1]的情况，不能用除法了，只能用乘法那个表达式，

所以，如果对于决策j1,j2满足上述表达式，则j2 优于 j1。

首先刚开始队首元素为0

用dy(i,j)表示[(f[i]-sum[i]+a[i+1]*i) - (f[j]-sum[j]+a[j+1]*j)]
用dx(i,j)表示(a[i+1]-a[j+1])*i
然后假设队列首尾指针head < tail 并且dy(queue[head],queue[head+1]) >=
i*dx(queue[head],queue[head+1])，那么队首元素直接丢掉就可以了。因为i是递
增的，如果当前queue[head]没有queue[head+1]好，那么今后也不会。
对于队尾的2个原素x, y来说，
如果对于当前i，y比x要差，那么由前面的证明：对于比较大的i，y不一定就比x差，
有可能比x好呢，我们来看看队尾3个元素的情况：x,y,z，如果
dy(x,y)/dx(x,y)>=dy(y,z)/dx(y,z)，那么可以直接把y给删了。因为
dy(x,y)/dx(x,y)和dy(y,z)/dx(y,z)是个常数，对于某个i，如果dy(x,y)/dx(x,y)<=i的
话，那么dy(y,z)/dx(y,z)一定也小于等于i，也就是说：如果y优于x，那么z一定优于
y，这个时候留着y就没用了。。。。直接删了。。。

过程就是这些,另外：因为有个限制k，所以决策点需要延迟加
入。
*/

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int MAXX = 500010;


int n, k, queue[MAXX];
__int64 sum[MAXX], f[MAXX], a[MAXX];

__int64 dy(int j1, int j2)
{
    return  (f[j1]-sum[j1]+a[j1+1]*j1) - (f[j2]-sum[j2]+a[j2+1]*j2);
}

__int64 dx(int j1, int j2)
{
    return  (a[j1+1] - a[j2+1]);
}

void dp()
{
    int i, j, head, tail, x, y, z;
    head = tail = 0;
    queue[0] = 0;
    for(i = 1; i <= n; i++)
    {
        while(head<tail && dy(queue[head], queue[head+1])>=i*dx(queue[head], queue[head+1]))
            head++;
        j = queue[head];
        f[i] = f[j] + sum[i] - sum[j] - a[j+1]*(i-j);
        if(i >= 2*k-1)    //实际上是i-k+1>=k
        {
            z = i-k+1;
            while(head < tail)
            {
                x = queue[tail-1];
                y = queue[tail];
                if(dy(x,y)*dx(y,z) >= dy(y,z)*dx(x,y))  tail--;
                else  break;
            }
            queue[++tail] = z;
        }
    }
}

int main()
{
    int t, i;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &k);
        sum[0] = 0;
        for(i = 1; i <= n; i++)
        {
            scanf("%I64d", a+i);
            sum[i] = sum[i-1] + a[i];
        }
        dp();
        printf("%I64d\n", f[n]);
    }
    return 0;
}

ok！