您的位置:首页 > 其它

BZOJ 1855 [Scoi2010]股票交易 单调队列优化DP

2015-06-08 11:38 302 查看

题意:链接

方法:单调队列优化DP

解析:噢又是一道情况很多的题,然而三种更新我又落下一种导致样例不过,后来看题解才恍然- -最SB的一种更新居然忘了。

状态好想f[i][j]代表前i天有j双袜子时的最大利润。

三种更新:

第一种:f[i][j]=max(f[i][j],f[i−1][j])f[i][j]=max(f[i][j],f[i-1][j]);(然而我忘了这一种)

第二种:买入f[i][j]=max(f[i][j],f[i−w−1][k]−(j−k)∗a[i].ap)(k>=j−a[i].as)f[i][j]=max(f[i][j],f[i-w-1][k]-(j-k)*a[i].ap)(k>=j-a[i].as);

第三种:卖出f[i][j]=max(f[i][j],f[i−w−1][k]+(k−j)∗a[i].bp)(k<=max(maxp,j+a[i].bs))f[i][j]=max(f[i][j],f[i-w-1][k]+(k-j)*a[i].bp)(k<=max(maxp,j+a[i].bs))

然后我们观察式子,第一种更新O(1)完成,第二和第三的时候如果枚举k的话复杂度承受不了,所以考虑怎么优化,显而易见第二三种是线性的,所以考虑到队列可不可行?于是整理表达式,发现可行,则按照f[i-w-1][k]+k*a[i].ap以及f[i-w-1][k]+k*a[i].bp维护递减就可以。

代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 2010
using namespace std;
int t,maxp,w;
int f

;
int q
;
struct node
{
    int ap,bp,as,bs;
}a
;
int main()
{
    scanf("%d%d%d",&t,&maxp,&w);
    for(int i=1;i<=t;i++)
    {
        scanf("%d%d%d%d",&a[i].ap,&a[i].bp,&a[i].as,&a[i].bs);
    }
    memset(f,-0x3f,sizeof(f));
    for(int i=1;i<=t;i++)
    {
        for(int j=0;j<=a[i].as;j++)f[i][j]=-a[i].ap*j;
        for(int j=0;j<=maxp;j++)f[i][j]=max(f[i][j],f[i-1][j]);
        if(i-w-1>=0)
        {
            int head=0,tail=0;
            for(int j=0;j<=maxp;j++)
            {
                while(head<tail&&q[head]<j-a[i].as)head++;
                while(head<tail&&f[i-w-1][j]+j*a[i].ap>=f[i-w-1][q[tail-1]]+q[tail-1]*a[i].ap)tail--;
                q[tail++]=j;
                if(head<tail)f[i][j]=max(f[i][j],f[i-w-1][q[head]]-(j-q[head])*a[i].ap);
            }
            head=0,tail=0;
            for(int j=maxp;j>=0;j--)
            {
                while(head<tail&&q[head]>j+a[i].bs)head++;
                while(head<tail&&f[i-w-1][j]+j*a[i].bp>=f[i-w-1][q[tail-1]]+q[tail-1]*a[i].bp)tail--;
                q[tail++]=j;
                if(head<tail)f[i][j]=max(f[i][j],f[i-w-1][q[head]]+(q[head]-j)*a[i].bp);
            }
        }
    }
    int ans=0;
    for(int i=0;i<=maxp;i++)
    {
        ans=max(ans,f[t][i]);
    }
    printf("%d\n",ans);
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: