Always Be Wise

그리디 알고리즘 : 주식(백준 11501번) 본문

알고리즘/백준

그리디 알고리즘 : 주식(백준 11501번)

bewisesh91 2021. 12. 1. 11:34
728x90

▶ 문제 : https://www.acmicpc.net/problem/11501

 

11501번: 주식

입력의 첫 줄에는 테스트케이스 수를 나타내는 자연수 T가 주어진다. 각 테스트케이스 별로 첫 줄에는 날의 수를 나타내는 자연수 N(2 ≤ N ≤ 1,000,000)이 주어지고, 둘째 줄에는 날 별 주가를 나타

www.acmicpc.net

##### 문제 #####
# 홍준이는 요즘 주식에 빠져있다. 
# 그는 미래를 내다보는 눈이 뛰어나, 
# 날 별로 주가를 예상하고 언제나 그게 맞아떨어진다.
# 매일 그는 아래 세 가지 중 한 행동을 한다.
# 1. 주식 하나를 산다.
# 2. 원하는 만큼 가지고 있는 주식을 판다.
# 3. 아무것도 안 한다.
# 홍준이는 미래를 예상하는 뛰어난 안목을 가졌지만, 
# 어떻게 해야 자신이 최대 이익을 얻을 수 있는지 모른다. 
# 따라서 당신에게 날 별로 주식의 가격을 알려주었을 때, 
# 최대 이익이 얼마나 되는지 계산을 해달라고 부탁했다.

##### 입력 #####
# 입력의 첫 줄에는 테스트 케이스 수를 나타내는 자연수 T가 주어진다. 
# 각 테스트 케이스 별로 첫 줄에는 날의 수를 나타내는 자연수 N이 주어지고, 
# 둘째 줄에는 날 별 주가를 나타내는 N개의 자연수들이 공백으로 구분되어 순서대로 주어진다. 
# 날 별 주가는 10,000 이하다.

##### 출력 #####
# 각 테스트 케이스 별로 최대 이익을 나타내는 정수 하나를 출력한다. 
# 답은 부호 있는 64bit 정수형으로 표현 가능하다.


▶ 접근 방법

그리디 알고리즘 문제 답게 가장 비싸게 팔 수 있는 금액을 max로 지정하고
그 금액 보다 작은 금액일 때는 무조건 사는 것으로 생각하여 문제를 해결하였다. 

 

▶ 풀이 코드

import sys

T = int(sys.stdin.readline().rstrip())

for _ in range(T):
    N = int(sys.stdin.readline().rstrip())
    stock_info = list(map(int, sys.stdin.readline().split()))
    max = 0
    result = 0
    for i in reversed(range(N)):
        if stock_info[i] > max :
            max = stock_info[i]
        else :
            result += max - stock_info[i]
    print(result)

 

▶ 관련 링크

2021.11.27 - [알고리즘] - 그리디 알고리즘(Greedy Algorithm)이란?

2021.11.27 - [알고리즘] - 그리디 알고리즘 : 동전 0(백준 11047번)

2021.11.27 - [알고리즘] - 그리디 알고리즘 : 잃어버린 괄호(백준 1541번)

2021.11.27 - [알고리즘] - 그리디 알고리즘 : 회의실 배정(백준 1931번)

2021.11.27 - [알고리즘] - 그리디 알고리즘 : 신입사원(백준 1946번)

2021.11.27 - [알고리즘] - 그리디 알고리즘 : 멀티탭 스케줄링(백준 1700번)

 

Comments