Python 코딩테스트

그리디: 백준 2828 파이썬

bgeun2 2021. 7. 13. 12:23
반응형

문제: https://www.acmicpc.net/problem/2828

 

2828번: 사과 담기 게임

상근이는 오락실에서 바구니를 옮기는 오래된 게임을 한다. 스크린은 N칸으로 나누어져 있다. 스크린의 아래쪽에는 M칸을 차지하는 바구니가 있다. (M<n) 플레이어는="" 게임을="" 하는="" 중에="" 바구니를="" <="" p=""> </n)>

www.acmicpc.net

풀이: 

N, M = map(int, input().split())

j = int(input())

apple = []

for _ in range(j):
    apple.append(int(input()))

move = 0

end = M     # 바구니의 오른쪽 끝 위치를 저장
start = 1   # 바구니의 왼쪽 끝 위치를 저장 

for i in range(j):
    # 바구니가 사과를 받을 수 있는 위치인 경우 
    if(end >= apple[i] and start <= apple[i]):
        continue
    # 사과가 바구니의 오른쪽에서 떨어지는 경우
    elif (end < apple[i]):
        move += apple[i] - end
        end = apple[i]
        start = apple[i] - (M - 1)
    # 사과가 바구니의 왼쪽에서 떨어지는 경우
    elif (start > apple[i]):
        move += start - apple[i]
        start = apple[i]
        end = apple[i] + (M - 1)

print(move)
반응형