AcWing 1058. 股票买卖 V

最新推荐文章于 2024-06-24 00:47:43 发布

最新推荐文章于 2024-06-24 00:47:43 发布

阅读量169

点赞数 2

分类专栏： AcWing算法基础

本文链接：https://blog.csdn.net/weixin_45884316/article/details/108859872

版权

AcWing算法基础专栏收录该内容

64 篇文章 6 订阅

订阅专栏

该博客介绍了一个使用动态规划算法解决股票交易问题的实例。给定一个股票价格数组，求解在一次交易（买入-卖出）后需冷冻一天的情况下，能获得的最大利润。输入包含股票价格数组，输出为最大利润。提供的C++代码实现了动态规划状态转移方程，通过维护三种状态（未持有股票、持有股票、冷冻期）来找到最优解。

摘要由CSDN通过智能技术生成

题目

给定一个长度为 N 的数组，数组中的第 i 个数字表示一个给定股票在第 i 天的价格。

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。

输入格式

第一行包含整数 N，表示数组长度。

第二行包含 N 个不超过 $10000$ 的正整数，表示完整的数组。

输出格式

输出一个整数，表示最大利润。

数据范围

$1≤N≤10^5$

输入样例：

5
1 2 3 0 2

输出样例：

样例解释

对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]，第一笔交易可得利润 2-1 = 1，第二笔交易可得利润 2-0 = 2，共得利润 1+2 = 3。

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5;
int n,k,f[N][3],a[N],ans;

int main(){
    cin>>n;
    for(register int i=1; i<=n; i++) 
    	cin>>a[i];
    memset(f,-0x3f,sizeof(f));
    f[0][2] = 0;

    for(register int i=1; i<=n; i++) 
    {
        f[i][0] = f[i-1][1]+a[i];
        f[i][1] = max(f[i-1][1],f[i-1][2]-a[i]);
        f[i][2] = max(f[i-1][0],f[i-1][2]);
    }
    int ans = max(f[n][0],f[n][2])
    cout<<ans;
    return 0;
}