题解 | 股票专题

一、只允许一笔交易 一天两个状态，由于只有一笔交易，买入时只能是-prices[i]

// 买
dp[i][0] = Math.max(dp[i-1][0],-prices[i]);
// 卖
dp[i][1] = Math.max(dp[i-1][1],dp[i-1][0]+prices[i]);

二、允许多笔交易 一天两个状态

// 买
dp[i][0] = Math.max(dp[i-1][0],dp[i-1][1]-prices[i]);
// 卖
dp[i][1] = Math.max(dp[i-1][1],dp[i-1][0]+prices[i]);

三、只允许两笔交易 5个状态，初始化特别重要

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param prices int整型一维数组
     * @return int整型
     */
    public int maxProfit (int[] prices) {
        // write code here
        int len = prices.length;
        if (len == 0) {
            return 0;
        }
        int dp[][] = new int[len][5];
        // 初始化特别重要
        Arrays.fill(dp[0], -10000);
        dp[0][0] = -prices[0];
        dp[0][4] = 0;
        // 0 买
        // 1 买+卖
        // 2 买+卖+买
        // 3 买+卖+买+卖
        // 4 没买没卖
        for (int i = 1; i < len; i++) {
            // 买
            dp[i][0] = Math.max(dp[i - 1][4] - prices[i], dp[i - 1][0]);
            // 买+卖
            dp[i][1] = Math.max(dp[i - 1][0] + prices[i], dp[i - 1][1]);
            // 买+卖+买
            dp[i][2] = Math.max(dp[i - 1][1] - prices[i], dp[i - 1][2]);
            // 买+卖+买+卖
            dp[i][3] = Math.max(dp[i - 1][2] + prices[i], dp[i - 1][3]);
            // 没买没卖
            dp[i][4] = 0;
            
        }
        return Math.max(dp[len-1][3], dp[len-1][4]);
    }
}