描述
假设有一个数组,它的第i个元素是一支给定的股票在第i天的价格。如果你最多只允许完成一次交易(例如,一次买卖股票),设计一个算法来找出最大利润。
样例
样例1
输入: [3, 2, 3, 1, 2]
输出: 1
说明:你可以在第三天买入,第四天卖出,利润是 2 - 1 = 1
算法思路:
创建一个dp数组用于保存每天可以获得的最大收益
只能在交易一次,那么每天可以获得的最大收益应该是当天的价格减去前几天最低的价格,然后从dp数组中找到最高的价格。
这里有一个小技巧,就是不用每次都从头到尾查找最少的价格,如果要查第i天可以获得的最大收益,只需要将第i-2天前的最少价格(计算第i-1天的最大收益时已经算过了)和第i-1天的价格比就好了,不然算法会超时。同样查找最大收益也只需要将第i天的价格和第i-1天之前的最大收益相比较就可以了。
通过这个小技巧可以看出来,完全没必要使用dp,贪心可以花费更少的空间得到这个题的解。
DP:
class Solution {
public:
/**
* @param prices: Given an integer array
* @return: Maximum profit
*/
int maxProfit(vector<int> &prices) {
// write your code here
int length=prices.size();
int *dp=new int[length+1];
dp[0]=0;
dp[1]=0;
int maxprice=dp[0];
int minprice=prices[0];
for(int i=2;i<=length;i++)
{
for(int j=i-1;j<i;j++)
{
minprice=min(prices[j],minprice);
}
dp[i]=prices[i-1]-minprice;
maxprice=max(dp[i],maxprice);
}
return maxprice;
}
};
对应的贪心算法只需要做一点儿小小的修改
class Solution {
public:
/**
* @param prices: Given an integer array
* @return: Maximum profit
*/
int maxProfit(vector<int> &prices) {
// write your code here
int length=prices.size();
int maxprice=0;
int minprice=prices[0];
int temp;
for(int i=2;i<=length;i++)
{
for(int j=i-1;j<i;j++)
{
minprice=min(prices[j],minprice);
}
temp=prices[i-1]-minprice;
maxprice=max(temp,maxprice);
}
return maxprice;
}
};