20546번 / 기적의 매매법

📕 문제

 

https://www.acmicpc.net/problem/20546

 

 

📄 풀이

 

1. 1월 14일의 자산은 (현금 + 1월 14일의 주가 × 주식 수)로 계산

-> 준현, 성민이의 잔여 현금, 주식 수 변수 각각 필요

->  1월 14일의 주가:  stock[-1] 로 구현

 

 

 

💻 내 코드

 

n = int(input())
stock = list(map(int, input().split()))

last = len(stock)
jun = 0
seong = 0

for i in range(last):
    # q는 주식 수
    q = n // stock[i] 
  
    if q > 0 :
        # 준현 주식 스타일
        n -= q * stock[i] 
        jun = n + q * stock[last-1]
        
    # 성민 주식 스타일
        if stock[i] > stock[i+1] and stock[i+1] > stock[i+2]:
            
            n -= q * stock[i+2] 
            seong = n + q * stock[i+2]
        elif stock[i] < stock[i+1] and stock[i+1] < stock[i+2]:
            n += q * stock[i+2] 
            seong = n + q * stock[i+2]
    


print(jun)
print(seong)

if jun > seong:
    print("BNP")
elif jun < seong:
    print("TIMING")
elif jun == seong:
    print("SAMESAME")

 

💻 답안

  n = int(input())
stock = list(map(int, input().split()))

# 준현과 성민의 초기 현금 및 주식 보유 상태
# jun_cash: 준현이  보유 현금
# jun_stock : 준현 보유 주식 개수 
jun_cash = n
jun_stock = 0

seong_cash = n
seong_stock = 0

# 주식 가격 변화 처리
for i in range(len(stock)):
    # 준현: 가진 돈으로 최대 주식 구매
    if jun_cash >= stock[i]:
        jun_stock += jun_cash // stock[i]
        jun_cash %= stock[i]

    # 성민: 3일 연속 상승/하락 확인 및 매매
    if i >= 3:
        # 3일 연속 하락 -> 전량 매수
        if stock[i - 3] > stock[i - 2] > stock[i - 1] > stock[i]:
            seong_stock += seong_cash // stock[i]
            seong_cash %= stock[i]
        # 3일 연속 상승 -> 전량 매도
        elif stock[i - 3] < stock[i - 2] < stock[i - 1] < stock[i]:
            seong_cash += seong_stock * stock[i]
            seong_stock = 0

# 마지막 날 자산 계산
jun_total = jun_cash + jun_stock * stock[-1]
seong_total = seong_cash + seong_stock * stock[-1]

# 결과 출력


if jun_total > seong_total:
    print("BNP")
elif jun_total < seong_total:
    print("TIMING")
else:
    print("SAMESAME")

 

📁 참고자료

 

☺️ 새로 알게 된 지식

 

1. 3일 연속 상승/ 하락 구현할 때

 

지피티에서 아래 처럼 알려줌.

단순하게 이렇게 구현할 수 도 있음

 

if i >= 3:
        # 3일 연속 하락 -> 전량 매수
        if stock[i - 3] > stock[i - 2] > stock[i - 1] > stock[i]:
            seong_stock += seong_cash // stock[i]
            seong_cash %= stock[i]
        # 3일 연속 상승 -> 전량 매도
        elif stock[i - 3] < stock[i - 2] < stock[i - 1] < stock[i]:
            seong_cash += seong_stock * stock[i]
            seong_stock = 0

 

 

 

😉 리뷰