2022-6-13 真题练习

MT10 股票交易日

 

描述

在股市的交易日中，假设最多可进行两次买卖(即买和卖的次数均小于等于 2 )，规则是必须一笔成交后进行另一笔(即买-卖-买-卖的顺序进行)。给出一天中的股票变化序列，请写一个程序计算一天可以获得的最大收益。请采用时间复杂度低的方法实现。

给定价格序列 prices 及它的长度 n ，请返回最大收益。

 

数据范围： 1 \le n \le 500 \1≤n≤500  ，1 \le prices_i \le 1000 \1≤pricesi​≤1000 

import java.util.*;

public class Stock {
    public int maxProfit(int[] prices, int n) {
        // write code here
        int[][][] dp=new int[n][2][3];
        for (int i=0;i<n;i++){
            for (int j=2;j>=1;j--){
                if (i==0){
                    dp[i][0][j]=0;
                    dp[i][1][j]=-prices[i];
                    continue;
                }
                dp[i][0][j]=Math.max(dp[i-1][0][j],dp[i-1][1][j]+prices[i]);
                dp[i][1][j]=Math.max(dp[i-1][1][j],dp[i-1][0][j-1]-prices[i]);
            }
        }
        return dp[n-1][0][2];
    }
}

思路：股票通用思路，dp[i][j][k] 第i天 j=0或1代表无或者有，k为交易次数。注意一个数的情况。