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)