买卖股票5

给定一个长度为 $N$ 的数组,数组中的第 $i$ 个数字表示一个给定股票在第 $i$ 天的价格。 设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):

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

输入格式 第一行包含整数 $N$,表示数组长度。 第二行包含 $N$ 个不超过 $10000$ 的正整数,表示完整的数组。 输出格式 输出一个整数,表示最大利润。 数据范围

$1 \le N \le 10^5$

输入样例:

5 1 2 3 0 2

输出样例:

3

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

解题代码

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#include <bits/stdc++.h>
using namespace std;
const int N =  100000+100 ;
int dp[N][3];
int a[N];


int main()
{
    int n ;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    //一开始 可以处于可以买入状态
    dp[0][1] = dp[0][0] = -999999;
    
    for (int i=1;i<=n;i++) {
        // 买入 => 卖出,卖出
        // 1 卖出1天【没卖出】,2 卖出>= 2天【卖出1天变化】,0 买入【没卖出】
        dp[i][2] = max(dp[i-1][1],dp[i-1][2]);
        dp[i][1] = dp[i-1][0] + a[i];
        dp[i][0] = max(dp[i-1][2] - a[i],dp[i-1][0]);
    }
    
    cout << max (dp[n][2],dp[n][1]) ;
    
    
    return 0;
}