题目描述:
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iii
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。示例 1:
输入: [3,3,5,0,0,3,1,4]
输出: 6
解释: 在第 4 天(股票价格 = 0)的时候买入,在第 6 天(股票价格 = 3)的时候卖出,这笔交易所能获得利润 = 3-0 = 3 。
随后,在第 7 天(股票价格 = 1)的时候买入,在第 8 天 (股票价格 = 4)的时候卖出,这笔交易所能获得利润 = 4-1 = 3 。
通过示例,首先进行初步编程,设置两次买入卖出赚的钱放入jy[2]数组中,设置当前所进行的是第flag次买入。当对每一个元素进行循环时,若后一个元素一直大于初始元素,则一直进行递增,直到取到最大的数,之后将该卖出的钱数放入jy中,数组中的元素已满,则进行比较留下赚钱最多的两个元素,之后再从下一个元素进行迭代。
int maxProfit(int* prices, int pricesSize){
int i=0,j,flag=0;
int jy[2]={0,0},sum;
while(i<pricesSize-1)
{
j=i+1;
if(prices[i]<prices[j])
{
flag++;
while(j+1<pricesSize&&prices[j]<prices[j+1]) {j++;}
sum=prices[j]-prices[i];
if(flag>=2)
{
if(jy[1]<sum)
{
jy[0]=jy[1];
jy[1]=sum;
}
else if(jy[0]<sum)
{
jy[0]=sum;
}
}
else if(flag==1) jy[1]=sum;
}
i=j;
}
return jy[1]+jy[0];
}
这样的方法只对较简单的数列有用,当数列变为:1 5 2 8 3 10 时,则不能用上述方法判断。从leetcode题解中学习了状态机方法(点击查看),并且参考了相应的c语言解法(点击查看)之后,最终得出正确代码,符合任意条件。
static inline
int max(int a,int b){
return a>b?a:b;
}
int maxProfit(int* prices, int pricesSize){
int dp_i10 = 0, dp_i11 = -32768;
int dp_i20 = 0, dp_i21 = -32768;
for (int i=0;i<pricesSize;i++) {
dp_i20 = max(dp_i20, dp_i21 + prices[i]);
dp_i21 = max(dp_i21, dp_i10 - prices[i]);
dp_i10 = max(dp_i10, dp_i11 + prices[i]);
dp_i11 = max(dp_i11, -prices[i]);
}
return dp_i20;
}