UVA 1427 Parade

来源:互联网 发布:sql语句count用法 编辑:程序博客网 时间:2024/05/21 05:17

单调队列优化DP

参考原文

题意: F城由n+1个横向路和m+1个竖向路组成。你的任务是从最南边的路走到最北边的路,使得走过的路上的高兴值的和最大(高兴值可能为负数)。同一段路最多只能走一次,且不能从北往南走。另外,在每条横向路上所花的时间不能超过k。1<=n<=100,1<=m<=10000,0<=k<=3000000。

解法: 很容易想到设dp[i][j]到第i行第j列那个点的最大值,设L[i][j]为从第i行的左边走到i的最大值,R[i][j]为从第i行的右边走到i的最大值,那么有如下转移:

dp[i][j] = max(L[i][j],R[i][j],dp[i+1][j])。注意边界问题。

乍看之下问题貌似解决了。但是呢,如果对于每个点都要对他的同一行扫一遍来计算他的L和R,要m的时间,总共约n*m个点,复杂度O(n*m^2),m范围有1W,不超时直播吃键盘- -!。

所以呢,考虑优化一下L和R的计算: L[i][j] = min(dp[i+1][k] + val[k~j]),0<=k<j。

如果happy值val处理成前缀和的形式,会不会更快一点呢,那么方程变为: L[i][j] = min(dp[i+1][k] + SumVal[0~j] - SumVal[0~k])。

光是前缀和貌似效果还不够好,那么方程再变个型: L[i][j] = min(dp[i+1][k] - SumVal[0~k]) + SumVal[0~j]。

注意min括号里的东西,dp[i+1][k]-SumVal[0~k],这个式子,跟上上上面的方程的区别在于,他只和k有关,k<j,那么,集合{0~j} - 集合{0~j-1} = {j},明显的递推关系!那么在知道前向的结果的情况下,计算出这一项只需要O(1)的时间!因此时间上可以优化到O(n*m)。

考虑对于横向路的时间不能超过k这个限制的处理。我们需要的值是j节点左端不超过k时间范围内的值,而每次递推时增加的值在时间上都是稳定最小的,天然的时间递减顺序!

考虑插入时间为k的这个值时,他必定是在最右端的,对于他左端比他小的值,可以直接忽略了(因为更小的范围,值还比你大,我要你做什么),而左端比他大的值,若时间不超过范围,则不能删去,因为可能会用到。这样,就构成了一个由左到右,在时间上递减,在happy值上也递减的序列,要维护这样一个序列,要支持两端删除,右端插入,显然是双端队列这个数据结构了。每次要取最大值时从最左端取值,时间O(1),每个值最多进队一次,出队一次,处理一行的时间为O(m),so nice!

总结: 单调队列优化的DP具有这样的性质: 有dp[i] = min(h[i]) + g[i],他的取最值部分h[i]可以被划分成与i值无关的,可递推的式子,这样的方程用单调队列能够将复杂度降一个幂次。

对于这道题,在POJ上貌似用C++光是读入就已经超时了。所以用G++和快速读入优化了一下,一不小心刷到第一了- -

/* **********************************************Author      : NeroCreated Time: 2013-8-30 15:17:38Problem id  : UVA 1427Problem Name: Parade*********************************************** */#include <stdio.h>#include <string.h>#include <algorithm>#include <ctype.h>using namespace std;#define REP(i,a,b) for(int i=(a); i<(int)(b); i++)#define clr(a,b) memset(a,b,sizeof(a))const int INF = 0x3f3f3f3f;                                                                                                                                   int mat[110][10100];int path[110][10100];int n,m,K;struct Q {    int val,time;}q[10100];int qf,qe;int dp[10100];int temp[10100];inline int GetInt() {    char c;    do {        c = getchar();    }while(!isdigit(c) && c != '-');    int ret = 0;    int sigma = 1;    if(c == '-') sigma = -1;    else ret = c - '0';    while(isdigit(c = getchar())) ret = ret * 10 + c - '0';    return ret*sigma;}int main() {    while(~scanf("%d%d%d", &n, &m, &K), n || m || K) {        clr(dp,0);        for(int i = 0; i <= n; i ++) path[i][0] = path[i][m+1] = 0;        for(int i = 0; i <= n; i ++) mat[i][0] = mat[i][m+1] = 0;        for(int i = 0; i <= n; i ++) {            for(int j = 1; j <= m; j ++) {                mat[i][j] = GetInt();                mat[i][j] += mat[i][j-1];            }        }        for(int i = 0; i <= n; i ++) {            for(int j = 1; j <= m; j ++) {                path[i][j] = GetInt();                path[i][j] += path[i][j-1];            }        }        for(int i = n; i >= 0; i --) {            for(int j = 0; j <= m; j ++) temp[j] = dp[j];            qf = 0;            qe = -1;            for(int j = 1; j <= m; j ++) {                int tv = temp[j-1] - mat[i][j-1];                while(qf <= qe && q[qe].val <= tv) qe --;                q[++qe].val = tv;                q[qe].time = path[i][j-1];                while(qf <= qe && path[i][j]-q[qf].time > K) qf ++;                if(qf <= qe) dp[j] = max(dp[j], q[qf].val+mat[i][j]);            }            qf = 0;            qe = -1;            for(int j = m-1 ; j >= 0; j --) {                int tv = temp[j+1] + mat[i][j+1];                while(qf <= qe && q[qe].val <= tv) qe --;                q[++qe].val = tv;                q[qe].time = path[i][j+1];                while(qf <= qe && q[qf].time - path[i][j] > K) qf ++;                if(qf <= qe) dp[j] = max(dp[j], q[qf].val-mat[i][j]);            }        }                int maxn = -(~0u>>1);        for(int i = 0; i <= m; i ++) {            maxn = max(maxn, dp[i]);        }        printf("%d\n", maxn);    }    return 0;}