最多n次购买时股票最大收益问题

一般我们常见的股票最大收益问题是：

已知一支股票连续n天的价格走势，以长度为n的整数数组表示，数组中第i个元素（prices[i]）代表该股票第i天的股价。 假设你一开始没有股票，但有至多两次买入1股而后卖出1股的机会，并且买入前一定要先保证手上没有股票。若两次交易机会都放弃，收益为0。 设计算法，计算你能获得的最大收益。 输入数值范围：2<=n<=100,0<=prices[i]<=100

这道算法比较容易理解的思想是：在对股票进行迭代遍历的过程中，用四个变量（第一、二次买入和卖出时的最大收益）记录当前的最大收益值，即记录当前的最优解。所以这道题可以用动态规划来解决。

针对这个问题的求解Java代码如下（带详细注释）：

public class Solution {
    /**
     * 计算你能获得的最大收益
     * 
     * @param prices Prices[i]即第i天的股价
     * @return 整型
     */
    public int calculateMax(int[] prices) {
        // 这四个变量表示每次买入和卖出时的当前最大收益。
        // 因为买入时，收益是减少的，减少的收益值为“当前买入股票的价格”，
        // 所以买入时收益的初始值为整数最小值
        int profitOfFirstBuy = Integer.MIN_VALUE, profitOfFirstSell = 0;
        int profitOfSecondBuy = Integer.MIN_VALUE, profitOfSecondSell = 0;

        for (int curPrice : prices) {

            // 第一次买入时的最大收益值，因为是第一次买入，所以其收益值是当前股票的负值，
            // 如果profitOfFirstBuy小于此值，则更新profitOfFirstBuy 
            if (profitOfFirstBuy < -curPrice) profitOfFirstBuy = -curPrice;

            //第一次卖出时的最大收益值，其收益值为当前股票值加上第一次买入时的收益值，
            //即当前股票值减去第一次买入的股票值
            if (profitOfFirstSell < curPrice + profitOfFirstBuy) profitOfFirstSell = curPrice + profitOfFirstBuy;

            //第二次买入时的最大收益值，假使当前作为第二次买入，则收益为第一次卖出时的收益减去当前股票值，
            //判断profitOfSecondBuy是否比此值小，如果小，则更新profitOfSecondBuy，
            //profitOfSecondBuy是记录第二次买入时收益的最大值
            if (profitOfSecondBuy < -curPrice + profitOfFirstSell) profitOfSecondBuy = -curPrice + profitOfFirstSell;

            // 第二次卖出时的最大收益值，第二次卖出最大收益为当前股票值加上第二次买入时的最大收益值。
            if (profitOfSecondSell < curPrice + profitOfSecondBuy) profitOfSecondSell = curPrice + profitOfSecondBuy;
        }

        // 因为最多可以买入两次，所以第二次卖出时的最大收益值即时所求的答案
        return profitOfSecondSell;
    }
}

因为代码的for循环中的核心代码是四条 比较值大小的判断语句，所以可以直接转换成使用Math.max()函数来进行处理，所以便有了这道题常见的求解代码：

    /**
     * 计算你能获得的最大收益
     *
     * @param prices Prices[i]即第i天的股价
     * @return 整型
     */
    public int calculateMax(int[] prices) {
        int profitOfFirstBuy = Integer.MIN_VALUE, profitOfFirstSell = 0;
        int profitOfSecondBuy = Integer.MIN_VALUE, profitOfSecondSell = 0;

        for (int curPrice : prices) {

            profitOfFirstBuy = Math.max(profitOfFirstBuy, -curPrice);
            profitOfFirstSell = Math.max(profitOfFirstSell, curPrice + profitOfFirstBuy);

            profitOfSecondBuy = Math.max(profitOfSecondBuy, -curPrice + profitOfFirstSell);
            profitOfSecondSell = Math.max(profitOfSecondSell, curPrice + profitOfSecondBuy);
        }
        return profitOfSecondSell;
    }

基于当前的思想，可以对这道题进行扩展，即扩展为“最多可以n次购买时股票的最大收益为题”，采用和上述同样的思想，java求解代码如下：

    /**
     * 计算你能获得的最大收益
     *
     * @param prices Prices[i]即第i天的股价
     * @param maxBuyTimes 最多的买卖次数
     * @return 整型
     */
    public int calculateMax(int[] prices, int maxBuyTimes) {
        int[] maxProfitOfBuy = new int[maxBuyTimes + 1];
        int[] maxProfitOfSell = new int[maxBuyTimes + 1];

        for (int i = 0; i <= maxBuyTimes; i++) {
            maxProfitOfBuy[i] = Integer.MIN_VALUE;
            maxProfitOfSell[i] = 0;
        }
        for (int curPrice : prices) {

            maxProfitOfBuy[0] = -curPrice;

            for (int i = 1; i <= maxBuyTimes; i++) {

                maxProfitOfBuy[i] = Math.max(maxProfitOfBuy[i], -curPrice + maxProfitOfSell[i - 1]);

                maxProfitOfSell[i] = Math.max(maxProfitOfSell[i], curPrice + maxProfitOfBuy[i]);
            }

        }

        return maxProfitOfSell[maxBuyTimes];
    }

注意：这个算法不保证第二次买入时间点在第一次卖出时间点之后。在迭代开始之初即在股票交易第一天，第一次买入和第二次买入均进行，而卖出则在此之后。所以计算股票收益最大值时，第n次买入的时间与第j次（j < n）次卖出的时间没有先后要求。