面试经典算法系列之数组/字符串4 -- 买卖股票的最佳时机

面试经典算法题36-买卖股票的最佳时机

LeetCode.121
公众号：阿Q技术站

问题描述

给定一个数组 prices ，它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。

你只能选择 某一天 买入这只股票，并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。

返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润，返回 0 。

示例 1：

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

示例 2：

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

思路

1. 初始化最低买入价格和最大利润：
   • 将第一天的股票价格设为最低买入价格。
   • 将最大利润初始化为0。
2. 从第二天开始遍历价格数组：
   • 如果当前价格比最低买入价格低，则更新最低买入价格为当前价格。
   • 如果当前价格与最低买入价格的差大于最大利润，则更新最大利润为当前价格与最低买入价格的差。
3. 返回最大利润：
   • 遍历完成后，返回最大利润即可。

图解

这道题最经典的方法应该还是动态规划，今天给大家用一个比较简单的方法讲解，用示例一进行一个流程图的讲解。

            +-----------------------------------------------+
            |                   开始执行程序                  |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |             初始化 minPrice = 7,               |
            |          maxProfit = 0，遍历数组                |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |    检查第2天价格 1 是否小于 minPrice（7）         |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |         更新 minPrice 为 1，继续遍历             |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |    检查第3天价格 5 是否小于 minPrice（1）         |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |           更新 maxProfit 为 5（5-1），          |
            |       继续检查下一天价格是否小于 minPrice         |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |    检查第4天价格 3 是否小于 minPrice（1）         |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |           继续检查下一天价格是否小于 minPrice     |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |    检查第5天价格 6 是否小于 minPrice（1）         |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |         更新 maxProfit 为 5（6-1），            |
            |       继续检查下一天价格是否小于 minPrice         |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |    检查第6天价格 4 是否小于 minPrice（1）         |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |           继续检查下一天价格是否小于 minPrice     |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |                遍历结束，返回最大利润 5           |
            +-----------------------------------------------+
                                 |
                                 v
            +-----------------------------------------------+
            |                   结束程序                     |
            +-----------------------------------------------+

参考代码

C++

#include <iostream>
#include <vector>

using namespace std;

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.empty()) {
            return 0; // 如果价格数组为空，返回0
        }

        int minPrice = prices[0]; // 初始化最低买入价格为第一天价格
        int maxProfit = 0; // 初始化最大利润为0

        for (int i = 1; i < prices.size(); ++i) {
            if (prices[i] < minPrice) {
                minPrice = prices[i]; // 更新最低买入价格
            } else if (prices[i] - minPrice > maxProfit) {
                maxProfit = prices[i] - minPrice; // 更新最大利润
            }
        }

        return maxProfit; // 返回最大利润
    }
};

int main() {
    vector<int> prices = {7, 1, 5, 3, 6, 4}; // 输入股票价格数组
    Solution solution;
    int result = solution.maxProfit(prices); // 计算最大利润
    cout << "最大利润：" << result << endl; // 输出结果

    return 0;
}

Java

import java.util.Arrays;

class Solution {
    public int maxProfit(int[] prices) {
        if (prices.length == 0) {
            return 0; // 如果价格数组为空，返回0
        }

        int minPrice = prices[0]; // 初始化最低买入价格为第一天价格
        int maxProfit = 0; // 初始化最大利润为0

        for (int i = 1; i < prices.length; ++i) {
            if (prices[i] < minPrice) {
                minPrice = prices[i]; // 更新最低买入价格
            } else if (prices[i] - minPrice > maxProfit) {
                maxProfit = prices[i] - minPrice; // 更新最大利润
            }
        }

        return maxProfit; // 返回最大利润
    }
}

public class Main {
    public static void main(String[] args) {
        int[] prices = {7, 1, 5, 3, 6, 4}; // 输入股票价格数组
        Solution solution = new Solution();
        int result = solution.maxProfit(prices); // 计算最大利润
        System.out.println("最大利润：" + result); // 输出结果
    }
}

Python

from typing import List

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        if not prices:
            return 0  # 如果价格数组为空，返回0

        min_price = prices[0]  # 初始化最低买入价格为第一天价格
        max_profit = 0  # 初始化最大利润为0

        for price in prices[1:]:
            if price < min_price:
                min_price = price  # 更新最低买入价格
            elif price - min_price > max_profit:
                max_profit = price - min_price  # 更新最大利润

        return max_profit  # 返回最大利润

# 测试代码
prices = [7, 1, 5, 3, 6, 4]  # 输入股票价格数组
solution = Solution()
result = solution.maxProfit(prices)  # 计算最大利润
print("最大利润：", result)  # 输出结果