最近在刷leetcode的DP,发现对于我而言难度有点大,特此记录刷题的过程,本文讲的是股票买卖的通用模板。
解题思路
解题的原理在这里查看,因此写的实在太好,这里就简要概括一下,建议看完上述参考连接后之后再看本文。
用一个三维数组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,坚持分享题目!