股票买卖问题（动态规划）

目录

一、股票买卖

输入格式

输出格式

思路：

代码：

二、股票买卖 II

输入格式

输出格式

思路：

代码：

三、股票买卖 III

预处理做法：

代码：

动态规划（大佬做法）：  原链接

代码：

---

---

---

一、股票买卖

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

如果你最多只允许完成一笔交易（即买入和卖出一支股票），设计一个算法来计算你所能获取的最大利润。

注意你不能在买入股票前卖出股票。

输入格式

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

第二行包含 N 个不大于 1e9 的正整数，表示完整的数组。

输出格式

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

---

思路：

这道题目就简单了，他只要一次最大，所以我们可以记录一下从1~n之间的最小和n~1之间的最大，找出相距最大的就行了。也就是直接划分成两个区间，前i个的最小和后i个的最大相减。

代码：

#include <bits/stdc++.h>

using namespace std;

const int N = 1e6 + 10;
#define long long int

int n, a[N];
int dp[N][2];

void solve()
{
    cin >> n;
    for (int i = 1; i <= n; i ++ ) cin >> a[i];
    
    dp[0][0] = 0x3f3f3f3f;
    for (int i = 1; i <= n; i ++ )
    {
        dp[i][0] = dp[i - 1][0];
        dp[i][0] = min(dp[i][0], a[i]);
    }
    
    dp[n][1] = a[n];
    for (int i = n - 1; i >= 1; i -- )
    {
        dp[i][1] = dp[i + 1][1];
        dp[i][1] = max(dp[i][1], a[i]);
    }
    
    int res = 0;
    for (int i = 1; i <= n; i ++ )
        res = max(res, dp[i][1] - dp[i][0]);
    
    cout << res << '\n';
}

int main()
{
    int T = 1;
    // cin >> T;
    while (T -- )
    {
        solve();
    }
    return 0;
}

二、股票买卖 II

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

设计一个算法来计算你所能获取的最大利润。你可以尽可能地完成更多的交易（多次买卖一支股票）。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

输入格式

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

第二行包含 N 个不大于 10000 的正整数，表示完整的数组。

输出格式

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

思路：

这道题目就是上一个的加强版了，不限制你交易的次数，问最大的利润。

开一个dp[N][2]数组，dp[i][0] 代表前 i 个数里面我手里没有票的最大利润，dp[i][1] 代表前 i 个数里面手里有票的最大利润，不难推出动态转换方程。

代码：

#include <bits/stdc++.h>

using namespace std;

const int N = 1e6 + 10;
#define long long int

int n, a[N];
int dp[N][2];

void solve()
{
    cin >> n;
    for (int i = 1; i <= n; i ++ ) cin >> a[i];
    
    dp[1][0] = 0;
    dp[1][1] = a[1] * -1;
    for (int i = 2; i <= n; i ++ )
    {
        dp[i][0] = max(dp[i - 1][1] + a[i], dp[i - 1][0]);
        dp[i][1] = max(dp[i - 1][0] - a[i], dp[i - 1][1]);
    }
    
    cout << dp[n][0] << '\n';
}

int main()
{
    int T = 1;
    // cin >> T;
    while (T -- )
    {
        solve();
    }
    return 0;
}

三、股票买卖 III

预处理做法：

我的想法是和第一个差不多，只是两个区间变成利润的最大和利润的最小，别的变化不大，看代码吧。

代码：

#include <bits/stdc++.h>

using namespace std;

const int N = 1e6 + 10;
#define long long int

int n, a[N];
int dp[N][4];

void solve()
{
    cin >> n;
    for (int i = 1; i <= n; i ++ ) cin >> a[i];
    
    dp[0][0] = 0x3f3f3f3f;
    for (int i = 1; i <= n; i ++ )
    {
        dp[i][0] = dp[i - 1][0];
        dp[i][0] = min(dp[i][0], a[i]);
        
        dp[i][2] = max(dp[i - 1][2], a[i] - dp[i][0]);
    }
    
    dp[n][1] = a[n];
    for (int i = n - 1; i >= 1; i -- )
    {
        dp[i][1] = dp[i + 1][1];
        dp[i][1] = max(dp[i][1], a[i]);
        
        dp[i][3] = max(dp[i + 1][3], dp[i][1] - a[i]);
    }
    
    int res = 0;
    for (int i = 1; i <= n; i ++ )
        res = max(res, dp[i][2] + dp[i][3]);
    
    cout << res << '\n';
}

int main()
{
    int T = 1;
    // cin >> T;
    while (T -- )
    {
        solve();
    }
    return 0;
}

动态规划（大佬做法）：  原链接

f[0][i]:从来没有买过股票的最大收益
f[1][i]:第一次买股票的最大收益
f[2][i]:卖掉一次股票的最大收益
f[3][i]:卖掉一次股票，又买了一次的最大收益
f[4][i]:卖掉两次股票的最大收益

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
typedef long long ll;
#define MAXN 200011
ll f[5][MAXN];
int main()
{
    memset(f,0xcf,sizeof f); // 负无穷
    ll n;
    scanf("%lld",&n);
    f[0][0]=0;
    for(ll i=1;i<=n;++i)
    {
        ll x;
        scanf("%lld",&x);
        f[0][i]=f[0][i-1];
        f[1][i]=std::max(f[1][i-1],f[0][i-1]-x);
        f[2][i]=std::max(f[2][i-1],f[1][i-1]+x);
        f[3][i]=std::max(f[3][i-1],f[2][i-1]-x);
        f[4][i]=std::max(f[4][i-1],f[3][i-1]+x);
    }
    printf("%lld",std::max(f[0][n],std::max(f[2][n],f[4][n])));
    return 0;
}