Koala - 9기/기초 알고리즘 스터디

[백준/Python] #16435 스네이크버드

future0610 2023. 2. 19. 23:24

문제

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

 

16435번: 스네이크버드

첫 번째 줄에 과일의 개수 N (1 ≤ N ≤ 1,000) 과 스네이크버드의 초기 길이 정수 L (1 ≤ L ≤ 10,000) 이 주어집니다. 두 번째 줄에는 정수 h1, h2, ..., hN (1 ≤ hi ≤ 10,000) 이 주어집니다.

www.acmicpc.net


Algorithm

스네이크버드의 길이보다 작거나 낮은 곳에 있는 과일이 있다면 그 과일을 fruits에 추가하고 스네이크버드의 길이를 1만큼 늘린다. 이 과정을 모든 과일의 높이가 스네이크버드의 길이보다 높은 곳에 위치할 때까지 반복한다.

 


 

 

Code

import sys
input = sys.stdin.readline

N, L = map(int, input().split())
H = list(map(int, input().split()))

while True:
    fruits = []
    prevH = H.copy()
    while True:
        if len(H) == 0:
            break
        h = H.pop()
        if h <= L:
            L += 1
        else:
            fruits.append(h)
    if sorted(prevH) == sorted(fruits):
        break
    H = fruits.copy()

print(L)