美團筆試題:股票交易日

在股市的交易日中,假設最多可進行兩次買賣(即買和賣的次數均小於等於2),規則是必須一筆成交後進行另一筆(即買-賣-買-賣的順序進行)。給出一天中的股票變化序列,請寫一個程序計算一天可以獲得的最大收益。請採用實踐複雜度低的方法實現。
給定價格序列prices及它的長度n,請返回最大收益。保證長度小於等於500。
測試樣例:

[10,22,5,75,65,80],6

返回:87

import java.util.Scanner;

public class MeituanTest8 {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner scanner = new Scanner(System.in);
        while (scanner.hasNext()) {
            int n = scanner.nextInt();
            int[] prices = new int[n];
            for (int i = 0; i < n; i++) {
                prices[i] = scanner.nextInt();
            }
            System.out.println(maxProfit(prices, n));
        }
    }
    public static int maxProfit(int[] prices, int n) {
        //第i天之前的最大收益
        int[] preMaxProfit = new int[n];
        //第i天之後的最大收益
        int[] postMaxProfit = new int[n];
        //總的最大收益
        int maxProfit = Integer.MIN_VALUE;
        //如果今天的價格減掉最小价格比截止到昨天的最大收益大,
        //就用今天的價格減去最小的價格;否則,用截止到昨天的最大收益
        int minBuy = prices[0];
        for(int i=1;i<n;i++){
            minBuy = Math.min(minBuy, prices[i]);
            preMaxProfit[i] = Math.max(preMaxProfit[i-1], prices[i] - minBuy);
        }
        //如果最大價格減掉今天價格比明天以後買入的最大收益大,
        //就用最大價格減掉今天的價格;否則,用明天以後買入的最大收益 
        int maxSell = prices[n-1];
        for(int i=n-2;i>=0;i--){
            maxSell = Math.max(maxSell, prices[i]);
            postMaxProfit[i] = Math.max(postMaxProfit[i+1], maxSell - prices[i]);
        }
         //總的最大利潤 
        for(int i=0;i<n;i++){
            maxProfit = Math.max(maxProfit, preMaxProfit[i]+postMaxProfit[i]);
        }
        return maxProfit;
    }
}
發佈了104 篇原創文章 · 獲贊 17 · 訪問量 20萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章