188 Best Time to Buy and Sell Stock IV

188 Best Time to Buy and Sell Stock IV

连接:https://leetcode.com/problems/best-time-to-buy-and-sell-stock-iv/
问题描述:
Say you have an array for which the ith element is the price of a given stock on day i.

Design an algorithm to find the maximum profit. You may complete at most k transactions.

Note:
You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).

Credits:
Special thanks to @Freezen for adding this problem and creating all test cases.

这个问题就是限定k次数的买卖股票的最大收益,其实是找数组中没有重叠部分的k个区间,使k个区间中末尾数字减去开头数字的和最大。解这个问题需要用动态规划。使用“局部最优和全局最优解法”。我们维护两种量,一个是当前到达第i天可以最多进行j次交易,最好的利润是多少(global[i][j]),另一个是当前到达第i天,最多可进行j次交易,并且最后一次交易在当天卖出的最好的利润是多少(local[i][j])。下面我们来看递推式,全局的比较简单,
global[i][j]=max(local[i][j],global[i-1][j]),
也就是去当前局部最好的,和过往全局最好的中大的那个(因为最后一次交易如果包含当前天一定在局部最好的里面,否则一定在过往全局最优的里面)。对于局部变量的维护,递推式是
local[i][j]=max(global[i-1][j-1]+max(diff,0),local[i-1][j]+diff),
也就是看两个量,第一个是全局到i-1天进行j-1次交易,然后加上今天的交易,如果今天是赚钱的话(也就是前面只要j-1次交易,最后一次交易取当前天),第二个量则是取local第i-1天j次交易,然后加上今天的差值(这里因为local[i-1][j]比如包含第i-1天卖出的交易,所以现在变成第i天卖出,并不会增加交易次数,而且这里无论diff是不是大于0都一定要加上,因为否则就不满足local[i][j]必须在最后一天卖出的条件了)。以上部分应用自一个博客:http://blog.csdn.net/linhuanmars/article/details/23236995,因为自己水平有限,等对动态规划理解更深,自己能讲清楚这个问题再来修改这里,争取自己讲清楚。
初级解法:

int max(int a,int b)
{
  return a>b? a:b;
}
int maxProfit(int k, int* prices, int pricesSize) 
{
    if(k<1||pricesSize<2) return 0;
    int profit=0;
    if(pricesSize*2<=k)
    {
       for(int i=1;i<pricesSize;i++)
      profit+=max(prices[i]-prices[i-1],0);
       return profit;
    }

    int **local=(int **)malloc(sizeof(int*)*k);
    int **global=(int**)malloc(sizeof(int*)*k);
    for(int i=0;i<k;i++)
    {
       local[i]=(int *)malloc(sizeof(int)*(pricesSize));
       global[i]=(int *)malloc(sizeof(int)*(pricesSize));
       local[i][0]=0;
       global[i][0]=0;
    }

    for(int i=1;i<pricesSize;i++)
    {
      int pro=prices[i]-prices[i-1];
      for(int j=0;j<k;j++)
      {
          if(j==0)
        local[j][i]=max(max(pro,0),local[j][i-1]+pro);
          else
       local[j][i]=max(global[j-1][i-1]+max(pro,0),local[j][i-1]+pro);
       global[j][i]=max(local[j][i],global[j][i-1]);
      }
    }

    profit=  global[k-1][pricesSize-1];

    for(int i=0;i<k;i++)
    {
       free(local[i]);
       free(global[i]);
    }
    free(local);
    free(global);
    return profit;
}

改进的解法:

int max(int a,int b)
    {
      return a>b? a:b;
    }
int maxProfit(int k, int* prices, int pricesSize) 
{
    if(k<1||pricesSize<2) return 0;
    int profit=0;
    if(pricesSize*2<=k)
    {
       for(int i=1;i<pricesSize;i++)
       profit+=max(prices[i]-prices[i-1],0);
       return profit;
    }

    int *local=(int *)malloc(sizeof(int)*k);
    int *global=(int*)malloc(sizeof(int)*k);
    local[0]=0;
    global[0]=0;
    for(int i=1;i<pricesSize;i++)
    {
      int pro=prices[i]-prices[i-1];
      for(int j=k-1;0<=j;j--)
      {
          if(j==0)
        local[j]=max(max(pro,0),local[j]+pro);
          else
       local[j]=max(global[j-1]+max(pro,0),local[j]+pro);
          global[j]=max(local[j],global[j]);
      }
    }

    profit=  global[k-1];
    free(local);
    free(global);
    return profit;
}

这里第二个解法提交到leetcode中时竟然是 Wrong Answer,
这里写图片描述
可是我实际在vs2012中测试时结果是24,不是32,可能是编译器标准不一致所致。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值