[알고리즘] BOJ 16435 스네이크버드

[BOJ] 16435 스네이크버드 문제 바로가기

📍 문제

스네이크버드는 뱀과 새의 모습을 닮은 귀여운 생물체입니다.

스네이크버드의 주요 먹이는 과일이며 과일 하나를 먹으면 길이가 1만큼 늘어납니다.

과일들은 지상으로부터 일정 높이를 두고 떨어져 있으며 i (1 ≤ i ≤ N) 번째 과일의 높이는 hi입니다.

스네이크버드는 자신의 길이보다 작거나 같은 높이에 있는 과일들을 먹을 수 있습니다.

스네이크버드의 처음 길이가 L일때 과일들을 먹어 늘릴 수 있는 최대 길이를 구하세요.

📍 입력

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

두 번째 줄에는 정수 h1, h2, ..., hN (1 ≤ hi ≤ 10,000) 이 주어집니다.

📍 출력

첫 번째 줄에 스네이크버드의 최대 길이를 출력합니다.

---

📍 풀이

• 입력받은 나무의 높이를 오름차순으로 정렬합니다.
• 가장 작은 나무의 높이부터 가장 큰 나무의 높이까지 for 반복문을 통해 차례로 스네이크버드의 길이와 비교합니다.
• 만약 나무의 높이가 스네이크버드보다 작거나 같다면 스네이크버드의 길이를 1 증가시킵니다.
• 만약 나무의 높이가 스네이크버드보다 크다면 반복문을 종료합니다.
• 모든 반복수행이 종료된 후 스네이크버드의 높이를 출력합니다.

✍ 코드

from sys import stdin
N, L = map(int,stdin.readline().split())
H = list(map(int,stdin.readline().split()))
H.sort() # 나무의 높이를 정렬
for h in H: # 높이가 작은 나무부터 반복
  if L >= h: # 만약 나무의 높이가 스네이크버드보다 작거나 같다면
    L+=1 # 스네이크버드 길이 + 1
  else: # 만약 나무의 높이가 스네이크버드보다 크다면
    break # 반복문 종료
print(L)