C语言
#define MAX_ONE(a,b) (a>b?a:b)
int maxProfit(int* prices, int pricesSize){
int (*dp)[2] = NULL;
int i;
dp = calloc(pricesSize * 2, sizeof(int));
dp[0][0] = 0;
dp[0][1] = -prices[0];
for (i = 1; i < pricesSize; i++) {
dp[i][0] = MAX_ONE(dp[i-1][0], prices[i]+dp[i-1][1]);
dp[i][1] = MAX_ONE(dp[i-1][0]-prices[i], dp[i-1][1]);
}
return dp[i-1][0];
}
注意点
- 注意
for{}
循环后结果下标的i
需要回退1,即dp[i-1][0]; for{}
循环应该从1开始进行遍历;- calloc分配空间需要注意不能使用
sizeof(int(*)[2])
;因为int(*)[2]
也是指针类型,在64位操作系统里长度为8,不满足我们需要的场景会导致越界错误;
思想
动态规划
每天的子问题分解
- 卖掉过去买入的股票,手上剩余0股票;
- 买入当天的股票,手上剩余1股票;
- 保持昨天已经买入的股票,手上剩余1股票;
- 保持昨天已经赚取的收益,不买入股票,手上剩余0股票;
根据子问题进行分类,手上0或1个股票分别对应两种可能进行动态规划设计;