动态规划——LeetCode121 买卖股票的最佳时机

给定一个数组，它的第 i 个元素是一支给定股票第 i 天的价格。如果你最多只允许完成一笔交易（即买入和卖出一支股票），设计一个算法来计算你所能获取的最大利润。

注意你不能在买入股票前卖出股票。

• 示例 1:

  输入: [7,1,5,3,6,4]
  输出: 5
  解释: 在第 2 天（股票价格 = 1）的时候买入，在第 5 天（股票价格 = 6）的时候卖出，最大利润 = 6-1 = 5 。
  注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格。

• 示例 2:

  输入: [7,6,4,3,1]
  输出: 0
  解释: 在这种情况下, 没有交易完成, 所以最大利润为 0。

class Solution {
    public int maxProfit(int[] prices) {
        //暴力方式 (199 / 200 个通过测试用例)，最后一个用例超时了！
        //若数组逆序，则返回0
        boolean flag = true;
        for(int i = 0; i < prices.length - 1; i++){
            if(prices[i] > prices[i+1])
                continue;
            else{
                flag = false;
                break;
            }
        }
        if(flag)
            return 0;
        
        //若数组是非逆序
        int[] memo = new int[prices.length];
        for(int i = 0; i < prices.length - 1; i++){ //买入股票与售出股票的差值
            memo[i] = prices[i] > sell(prices, i) ? 0 : sell(prices, i) - prices[i]; //求差值
        }
        
        //找memo中的最大值
        int max = -1;
        for(int i = 0; i < memo.length; i++){
            if(memo[i] > max)
                max = memo[i];
        }
        return max;
    }
    //售卖股票，找最大值
    public int sell(int[] prices, int index){
        int [] arr = new int[prices.length-index-1];
        for(int i = 0; i < arr.length; i++)
            arr[i] = prices[i+index+1];
        Arrays.sort(arr);
        return arr[prices.length-index-2];
    }
}

• 动态规划优化

//动态规划优化
class Solution {
    public int maxProfit(int[] prices) {
        //若数组逆序，则返回0
        boolean flag = true;
        for(int i = 0; i < prices.length - 1; i++){
            if(prices[i] > prices[i+1])
                continue;
            else{
                flag = false;
                break;
            }
        }
        if(flag)
            return 0;
        
        //初始化dp
        int[] dp = new int[prices.length];  //dp值为盈利, 即保存正值, 将亏损的赋值为0;
        if(prices[prices.length-1] < prices[prices.length-2])
            dp[prices.length-2] = 0;
        else
            dp[prices.length-2] = prices[ prices.length - 1] - prices[prices.length-2];
        
        for(int i = prices.length - 2; i >= 1; i--){
            dp[i-1] = dp[i] + prices[i] - prices[i-1] > 0 ? dp[i] + prices[i] - prices[i-1] : 0;
        }
        //找最大值
        int max = -1;
        for(int i = 0; i < dp.length; i++){
            if(dp[i] > max)
                max = dp[i];
        }
        return max;
    }
}

执行用时 : 3 ms, 在Best Time to Buy and Sell Stock的Java提交中击败了75.57% 的用户
内存消耗 : 37.2 MB, 在Best Time to Buy and Sell Stock的Java提交中击败了83.34% 的用户
进行下一个挑战：