본문 바로가기

LeetCode

[LeetCode] - 121. Best Time to Buy and Sell Stock

LeetCode는 프로그래밍 문제를 풀며 코딩 실력을 향상할 수 있는 온라인 플랫폼입니다. 다양한 알고리즘 및 데이터 구조 문제를 제공하며, 면접 대비에 유용합니다. 해당 문제는, LeetCode Problems에서 볼 수 있는 난이도 '쉬움 (Easy)' 단계인 "Best Time to Buy and Sell Stock" 문제입니다.

--> https://leetcode.com/problems/best-time-to-buy-and-sell-stock/description/

 

문제 : 

You are given an array prices where prices[i] is the price of a given stock on the ith day. You want to maximize your profit by choosing a single day to buy one stock and choosing a different day in the future to sell that stock.

Return the maximum profit you can achieve from this transaction.

If you cannot achieve any profit, return 0.


Example 1 :

Input : prices = [7,1,5,3,6,4]

Output : 5

Explanation : Buy on day 2 (price = 1) and sell on day 5 (price = 6), profit = 6-1 = 5.

Note that buying on day 2 and selling on day 1 is not allowed because you must buy before you sell.

 

Example 2 :

Input : prices = [7,6,4,3,1]

Output : 0

Explanation : In this case, no transactions are done and the max profit = 0.

 

Constraints :

  • 1 <= prices.length <= 10^5
  • 0 <= prices[i] <= 10^4

이 문제에서는 주어진 배열에서 주식의 가격이 날마다 주어집니다. 주식을 사고파는 날짜를 선택하여 최대 이익을 얻고자 합니다. 단, 이익을 얻을 수 없다면 0을 반환합니다. 이 문제가 흥미로운 이유는 바로 Dynamic Programming, 동적 프로그래밍을 써야 하기 때문입니다. 

class Solution {
    public int maxProfit(int[] prices) {
        // 10^8 == 1 sec == O(n)
        int buyV = prices[0];
        int max_profit = 0;

        for (int p : prices) {
            buyV = Math.min(buyV, p);
            max_profit = Math.max(max_profit, p - buyV);
        }
        return max_profit;
    }
}

 

제 코드는 : 

  1. 초기화:
    • buyV는 현재까지의 최소 매수 가격을 저장합니다. 처음에는 배열의 첫 번째 값을 사용합니다.
    • max_profit은 현재까지의 최대 이익을 저장합니다. 초기 값은 0입니다.
  2.  
  3. 배열 순회 및 계산:
    • 배열을 순회하며 현재 주식 가격 p가 buyV보다 작으면 buyV를 갱신합니다.
    • 현재 주식 가격 p에서 buyV를 뺀 값을 계산하여 max_profit을 갱신합니다. 이는 현재 주식 가격에서 매수 가격을 뺀 값으로, 현재까지의 최대 이익을 의미합니다.
  4. 결과 반환:
    • 최대 이익을 반환합니다.

시간 복잡도는 O(n) 입니다.


Runtime 결과