HDU 3401 Trade【单调队列+dp】

本文探讨了在有限资金和交易限制下,如何通过优化买入和卖出时机来实现股票投资的最大化收益。通过动态规划算法,实现了在n天内根据价格波动、买入上限和卖出上限,制定最优交易策略。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题意:已知股票每天的买入和卖出价格、买入上限和卖出上限以及最多能持有的股票数,问n天后的最大收益是多少。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>

using namespace std;

const int oo = 1 << 30;
const int maxn = 2010;
int tcase;
int t, maxp, w;
int ap[maxn], bp[maxn];
int as[maxn], bs[maxn];
int dp[maxn][maxn];
int head, tail;
int q[maxn][2];

void DP()
{
    for (int i = 0; i < maxn; ++i) {
        for (int j = 0; j < maxn; ++j) {
            dp[i][j] = -oo;
        }
    }
    // 前w+1天只能买入 
    for (int i = 1; i <= w + 1; ++i) {
        for (int j = 0; j <= min(maxp, as[i]); ++j) {
            dp[i][j] = -ap[i] * j;
        }
    }
    for (int i = 2; i <= t; ++i) {
        /*for (int j = 0; j <= maxp; ++j) {
            dp[i][j] = max(dp[i][j], dp[i-1][j]);
        }
        if (i <= w + 1) continue;
        */
        head = tail = 0;
        for (int j = 0; j <= maxp; ++j) {
            dp[i][j] = max(dp[i][j], dp[i-1][j]);
            if (i <= w + 1) continue;
            while (head < tail && q[tail-1][1] <= dp[i-w-1][j] + ap[i] * j) {
                tail--;
            }
            q[tail][0] = j;
            q[tail++][1] = dp[i-w-1][j] + ap[i] * j;
            while (head < tail && j - q[head][0] > as[i]) {
                head++;
            }
            dp[i][j] = max(dp[i][j], q[head][1] - ap[i] * j);
        }
        head = tail = 0;
        for (int j = maxp; j >= 0; --j) {
            if (i <= w + 1) continue;
            while (head < tail && q[tail-1][1] <= dp[i-w-1][j] + bp[i] * j) {
                tail--;
            }
            q[tail][0] = j;
            q[tail++][1] = dp[i-w-1][j] + bp[i] * j;
            while (head < tail && q[head][0] - j > bs[i]) {
                head++;
            }
            dp[i][j] = max(dp[i][j], q[head][1] - bp[i] * j);
        }
    }
    int ans = -oo;
    for (int i = 0; i <= maxp; ++i) {
        ans = max(ans, dp[t][i]);
    }
    printf("%d\n", ans);
}

int main()
{
    scanf("%d", &tcase);
    while (tcase--) {
        scanf("%d%d%d", &t, &maxp, &w);
        for (int i = 1; i <= t; ++i) {
            scanf("%d%d%d%d", &ap[i], &bp[i], &as[i], &bs[i]);
        }
        DP();
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值