码迷,mamicode.com
首页 > 其他好文 > 详细

【bzoj1855||hdu3401】股票交易——单调队列优化dp

时间:2017-06-11 21:18:18      阅读:254      评论:0      收藏:0      [点我收藏+]

标签:style   卖出   display   return   原理   pac   利用   font   namespace   

代码基本是跟着yy dalao码的吧,

主要是要知道单调队列优化要怎么做,

f[i][j]表示第i天手里有j股时的收益。

从第一天到第w+1天初始化为-e[i].ap*j(因为此时只能买不能卖),其余为-inf。

分三种情况:

1.不买不卖:即f[i][j]=max(f[i][j],f[i-1][j]);

2.买进: f[i][j]=max(f[i-W-1][k]-(j-k)*e[i].ap,f[i][j])。在这里j-e[i].as<k<=mxa,利用f[i][j]=max(f[i-W-1][k]+k*e[i).ap-j*e[i].ap。令a[i-W-1][k]=f[i-W-1][k]+k*e[i].ap(事实上完全不用a这个数组,用单调队列),则f[i][j]=max(a[i-W-1][k]) - j*e[i].ap。枚举要从0到mxa

3.卖出,原理同上,枚举要从mxa到0

最后的f[n][0]即为答案(现实意义为第n天此时手里没有任何股票时的最大收益)。

具体实现看代码:

技术分享
#include<cstdio>
#include<cstring>
#include<iostream>
const int M=2010,inf=0x3f3f3f3f;
using namespace std;
struct point
{
    int ap,bp,as,bs;
}e[M];
struct node{int pos,k;}a[M];
int f[M][M],head,tail;
int main()
{
    int n,mxa,w;
    bool ok;
    memset(f,-inf,sizeof(f));
    scanf("%d %d %d",&n,&mxa,&w);
    for(int i=1;i<=n;i++)scanf("%d %d %d %d",&e[i].ap,&e[i].bp,&e[i].as,&e[i].bs);
    for(int i=1;i<=w+1;i++)
        for(int j=1;j<=e[i].as;j++)
        f[i][j]=-e[i].ap*j;
    f[0][0]=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<=mxa;j++)f[i][j]=max(f[i][j],f[i-1][j]);
        if(i<=w+1)continue;
        int now=i-w-1;
        head=tail=0;
        for(int j=0;j<=mxa;j++)
        {
            ok=true;
            int now1=f[now][j]+j*e[i].ap;
            while(head<tail&&a[tail-1].k<now1)tail--;
            a[tail].k=now1;a[tail++].pos=j;
            while(a[head].pos+e[i].as<j){if(head==tail){ok=false;break;}head++;}
            if(ok)f[i][j]=max(f[i][j],a[head].k-j*e[i].ap);
        }
        head=tail=0;
        for(int j=mxa;j>=0;j--)
        {
            ok=true;
            int now1=f[now][j]+j*e[i].bp;
            while(head<tail&&a[tail-1].k<now1)tail--;
            a[tail].k=now1;a[tail++].pos=j;
            while(a[head].pos-e[i].bs>j){if(head==tail){ok=false;break;}head++;}
            if(ok)f[i][j]=max(f[i][j],a[head].k-j*e[i].bp);
        }
    }
    printf("%d",f[n][0]);
    return 0;
}
bzoj1855

注意一点就是hdu3401这道题有多组数据,记得改一下再交啊(不然就像我一样WA了……)

技术分享
#include<cstdio>
#include<cstring>
#include<iostream>
const int M=2010,inf=0x3f3f3f3f;
using namespace std;
struct point
{
    int ap,bp,as,bs;
}e[M];
struct node{int pos,k;}a[M];
int f[M][M],head,tail;
int main()
{
    int n,mxa,w,tt;
    bool ok;
    scanf("%d",&tt);
    while(tt--)
    {
        memset(f,-inf,sizeof(f));
        scanf("%d %d %d",&n,&mxa,&w);
        for(int i=1;i<=n;i++)scanf("%d %d %d %d",&e[i].ap,&e[i].bp,&e[i].as,&e[i].bs);
        for(int i=1;i<=w+1;i++)
            for(int j=1;j<=e[i].as;j++)
            f[i][j]=-e[i].ap*j;
        f[0][0]=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=0;j<=mxa;j++)f[i][j]=max(f[i][j],f[i-1][j]);
            if(i<=w+1)continue;
            int now=i-w-1;
            head=tail=0;
            for(int j=0;j<=mxa;j++)
            {
                ok=true;
                int now1=f[now][j]+j*e[i].ap;
                while(head<tail&&a[tail-1].k<now1)tail--;
                a[tail].k=now1;a[tail++].pos=j;
                while(a[head].pos+e[i].as<j){if(head==tail){ok=false;break;}head++;}
                if(ok)f[i][j]=max(f[i][j],a[head].k-j*e[i].ap);
            }
            head=tail=0;
            for(int j=mxa;j>=0;j--)
            {
                ok=true;
                int now1=f[now][j]+j*e[i].bp;
                while(head<tail&&a[tail-1].k<now1)tail--;
                a[tail].k=now1;a[tail++].pos=j;
                while(a[head].pos-e[i].bs>j){if(head==tail){ok=false;break;}head++;}
                if(ok)f[i][j]=max(f[i][j],a[head].k-j*e[i].bp);
            }
        }
        printf("%d\n",f[n][0]);
    }
    return 0;
}
hdu3401

 

【bzoj1855||hdu3401】股票交易——单调队列优化dp

标签:style   卖出   display   return   原理   pac   利用   font   namespace   

原文地址:http://www.cnblogs.com/JKAI/p/6986213.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!