题目如下:
假设你有一个数组prices,长度为n,其中prices[i]是股票在第i天的价格,请根据这个价格数组,返回买卖股票能获得的最大收益
1.你可以买入一次股票和卖出一次股票,并非每天都可以买入或卖出一次,总共只能买入和卖出一次,且买入必须在卖出的前面的某一天
2.如果不能获取到任何利润,请返回0
3.假设买入卖出均无手续费
要求:空间复杂度 O(1),时间复杂度 O(n)
已有给定prices数组:vector<int>& prices
首先创建一个dp[n]数组。dp[i]表示到i天时候卖出(天数从0
开始编号)所能取得的最大收益。
int n=prices.size();
vector<int> dp(n,0);
光简单描述很难说明dp数组的含义,我举个例子方便理解:假设prices数组的值为:[1,2,3,4,5],那么dp[1]=1:即以1的单价买入,2的单价卖出。dp[2]=2,注意这里的理解不是第一天买入第三天卖出,而是第一天买入第二天卖出的基础上第二天再买入第三天卖出 。
此时dp[2]=dp[1]+prices[2]-prices[2-1]=2。
推理可知如果dp[i]>0有公式: dp[i]=dp[i-1]+prices[i]-prices[i-1];//这种写法相当于消除中间变量
再举例假设prices数组的值为:[8,9,2,5,4,7,1] 那么dp[1]=第一天买入第二天卖出的收益dp[1]=dp[0]+prices[1]-prices[0]=1,dp[2]表示的就是第二天卖出股票能获得的最大收益,此时dp[2]=dp[1]+prices[2]-prices[2-1]。此时dp[2]<0,dp突然变得小于零,则第i天的票价比已知的最小值还要小,所以要重新赋值那么dp[3]的需要重新初始化为prices[3]-prices[2]=5。并且我们需要一个Max来获取dp数组中的最大值。
if(dp[i-1]<0){
dp[i]=prices[i]-prices[i-1];
}
两种情况讨论完毕,接下来给出函数代码
int maxProfit(vector<int>& prices) {
// write code here
int n=prices.size();
vector<int> dp(n,0);
int Max=0;
if (n<=1) return 0;
for(int i=1;i<n;i++)
{
if(dp[i-1]<0){
dp[i]=prices[i]-prices[i-1];
}
else{
dp[i]=dp[i-1]+prices[i]-prices[i-1];
}
Max=max(dp[i],Max);
}
return Max;
}
输入prices数组的值为:[8,9,2,5,4,7,1],dp数组的值为:[0,1,-6,3,2,5,-1].Max取5。
讲的有点模糊,如有不懂欢迎提问。抱歉,头脑想的出来但是表述的不是很好,