LeetCode 股票买卖问题模板 (121 、122、123、188、309、714题)

最近在刷leetcode的DP，发现对于我而言难度有点大，特此记录刷题的过程，本文讲的是股票买卖的通用模板。

• 股票买卖问题1
• 股票买卖问题2
• 股票买卖问题3
• 股票买卖问题4
• 股票买卖问题含冰冻期
• 股票买卖问题含手续费

解题思路

解题的原理在这里查看，因此写的实在太好，这里就简要概括一下，建议看完上述参考连接后之后再看本文。

用一个三维数组dp[i][k][s]来表示所有的状态的最大收益，其中i表示天数，k表示最多允许交易数，s=0表示此时没有股票，s=1表示有股票。例如dp[3][4][1]就表示第3天，最多允许交易4次，有股票在手上时的最大收益。

DP最主要的就是找到状态转移方程与边界条件，这里也直接给出。
其中状态转移方程为：

dp[i][k][0]=MAX(dp[i-1][k][0],dp[i-1][k][1]+prices[i])
//简单可以理解为：要想今天没有股票，有两个方法(即可以由两个状态转移而来)：
//1.昨天就没有股票，今天又没有购入。
//2.昨天有股票，今天卖出了。
//两者取最大值就是今天的最大利润。

dp[i][k][1]=MAX(dp[i-1][k][1],dp[i-1][k-1][0]-prices[i])
//简单可以理解为：要想今天有股票，有两个方法(即可以由两个状态转移而来)：
//1.昨天就有股票，今天没有卖出。
//2.昨天没有股票，今天买入。
//两区取最大值就是今天的最大利润。

基础条件为:

dp[0][k][0]=0;
dp[0][k][1]=-prices[0];

因此给出股票问题的模板：

#define MAX(a, b) ((a) > (b) ? (a) : (b))

int maxProfit(int* prices, int pricesSize,int k){//有的题目k已经给出。
	int i,j;
    int dp[100001][k+1][2]={0};  //设置足够大的天数n即可。
    if(pricesSize==0)return 0; //数组长度为0的时候返回0
    for(i=0;i<pricesSize;i++){
        if(i==0){              //初始化基础条件
            for(j=0;j<=k;j++){
                dp[0][j][0]=0;
                dp[0][j][1]=-prices[0];
            }
            continue;
        }
        for(j=k;j>=1;j--){          // 动态转移方程
            dp[i][j][0]=MAX(dp[i-1][j][0],dp[i-1][j][1]+prices[i]);
            dp[i][j][1]=MAX(dp[i-1][j][1],dp[i-1][j-1][0]-prices[i]);
        }
    }
    return dp[pricesSize-1][k][0];   //返回dp[len-1][k][0]
}

现在来套模板做题：

股票问题1

分析题目：其中k=1。

#define MAX(a, b) ((a) > (b) ? (a) : (b))

int maxProfit(int* prices, int pricesSize){
    int i,j,k=1;
    int dp[100001][2][2]={0};  //设置足够大的天数n即可。
    if(pricesSize==0)return 0; //数组长度为0的时候返回0
    for(i=0;i<pricesSize;i++){
        if(i==0){              //初始化基础条件
            for(j=0;j<=k;j++){
                dp[0][j][0]=0;
                dp[0][j][1]=-prices[0];
            }
            continue;
        }
        for(j=k;j>=1;j--){          // 动态转移方程
            dp[i][j][0]=MAX(dp[i-1][j][0],dp[i-1][j][1]+prices[i]);
            dp[i][j][1]=MAX(dp[i-1][j][1],dp[i-1][j-1][0]-prices[i]);
        }
    }
    return dp[pricesSize-1][k][0];   //返回dp[len-1][k][0]
}

股票问题2

本题中k为无穷，可以直接将[k]省略，即

dp[i][k][s]->dp[i][s]
//状态转移方程为
dp[i][0]=MAX(dp[i-1][0],dp[i-1][1]+prices[i]);
dp[i][1]=MAX(dp[i-1][1],dp[i-1][0]-prices[i]);
//基础条件为
dp[0][0]=0;
dp[0][1]=-prices[0];
//其实区别不太。

直接给出源码

#define MAX(a,b) ((a)>(b)?(a):(b))

int maxProfit(int* prices, int pricesSize){
    int i,k=1,j;
    int dp[100001][2]={0};  //省略掉k
    if(pricesSize==0)return 0;
    for(i=0;i<pricesSize;i++){
        if(i==0){
            for(j=0;j<=k;j++){ //此处可以不要k，但为了和模板向匹配，这里给出没用的循环。
                dp[0][0]=0;
                dp[0][1]=-prices[0];
            }
            continue;
        }
        for(j=k;j>=1;j--){
            dp[i][0]=MAX(dp[i-1][0],dp[i-1][1]+prices[i]);
            dp[i][1]=MAX(dp[i-1][1],dp[i-1][0]-prices[i]);
        }
    }
    return dp[pricesSize-1][0];
}

股票问题3

分析题目，其中k=2。

#define MAX(a,b) ((a)>(b)?(a):(b))

int maxProfit(int* prices, int pricesSize){
    int i,k=2,j;
    int dp[100001][3][2]={0};  //中间的数字为k
    if(pricesSize==0)return 0;
    for(i=0;i<pricesSize;i++){
        if(i==0){
            for(j=0;j<=k;j++){
                dp[0][j][0]=0;
                dp[0][j][1]=-prices[0];
            }
            continue;
        }
        for(j=k;j>=1;j--){
            dp[i][j][0]=MAX(dp[i-1][j][0],dp[i-1][j][1]+prices[i]);
            dp[i][j][1]=MAX(dp[i-1][j][1],dp[i-1][j-1][0]-prices[i]);
        }
    }
    return dp[pricesSize-1][k][0];
}

股票问题4

本题中k用参数形式给出。

#define MAX(a,b) ((a)>(b)?(a):(b))

int maxProfit(int k, int* prices, int pricesSize){
    int i,j;
    int dp[100001][k+1][2];
    if(pricesSize==0)return 0;
    for(i=0;i<pricesSize;i++){
        if(i==0){
            for(j=0;j<=k;j++){
                dp[0][j][0]=0;
                dp[0][j][1]=-prices[0];
            }
            continue;
        }
        for(j=k;j>=1;j--){
            dp[i][j][0]=MAX(dp[i-1][j][0],dp[i-1][j][1]+prices[i]);
            dp[i][j][1]=MAX(dp[i-1][j][1],dp[i-1][j-1][0]-prices[i]);
        }
    }
    return dp[pricesSize-1][k][0];
}

到这里，可以看出，上面几个题目，其实就是围绕k来做变化的。

下面还有两个变式。

股票问题含冰冻期

所谓冰冻期就是在股票买入的时候考虑i-2天时的情况。

//动态转移方程变为：
dp[i][k][0]=MAX(dp[i][k-1][0],dp[i-1][k][1]+prices[i])
dp[i][k][1]=MAX(dp[i][k-1][1],dp[i-2][k][1]+prices[i])
//基础条件没变。

本题的源码为（其实就是用变量记录dp[i-2][k][0]）：

#define MAX(a,b) ((a)>(b)?(a):(b))
int maxProfit(int* prices, int pricesSize){
    int i,k=1,j,p=0,temp;
    int dp[100001][2][2]={0};
    if(pricesSize==0)return 0;
    for(i=0;i<pricesSize;i++){
        if(i==0){
            for(j=0;j<=k;j++){
                dp[0][j][0]=0;
                dp[0][j][1]=-prices[0];
            }
            continue;
        }
        for(j=k;j>=1;j--){
            temp=dp[i-1][j][0];
            dp[i][j][0]=MAX(dp[i-1][j][0],dp[i-1][j][1]+prices[i]);
            dp[i][j][1]=MAX(dp[i-1][j][1],p-prices[i]);
            p=temp;
        }
    }
    return dp[pricesSize-1][k][0];
}

股票问题加手续费

所谓手续费就是在股票卖出的时候减去手续费。

其中未规定k的大小，当做无穷，可以省略，和股票问题2相同。

#define MAX(a,b) ((a)>(b)?(a):(b))

int maxProfit(int* prices, int pricesSize, int fee){
    int dp[100001][2]={0};
    int i,k=1,j;
    if(pricesSize==0)return 0;

    for(i=0;i<pricesSize;i++){
        if(i==0){
            for(j=0;j<=k;j++){  //此处可以省略k循环
                dp[0][0]=0;
                dp[0][1]=-prices[0];
            }
            continue;
        }
        dp[i][0]=MAX(dp[i-1][0],dp[i-1][1]+prices[i]-fee);
        dp[i][1]=MAX(dp[i-1][1],dp[i-1][0]-prices[i]);
    }
    return dp[pricesSize-1][0];
}

到此，leetcode中的股票问题已经结束，相信你已经会用了。

许久未写博客，愿能坚持刷LeetCode，坚持分享题目！