[Silver V] 스네이크버드 - 16435 (그리디, 정렬) : JAVA, python

[Silver V] 스네이크버드 - 16435

문제 링크

성능 요약

메모리: 31256 KB, 시간: 40 ms

분류

그리디 알고리즘(greedy), 정렬(sorting)

문제 설명

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

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

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

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

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

입력

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

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

출력

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

풀이 출력

뱀의 길이보다 과일 높이가 작거나 같을 때만 먹을 수 있고, 먹을 때마다 뱀의 길이가 1씩만 늘어나기 때문에 뱀의 최대 길이를 구하기 위해서는 과일 높이를 오름차순으로 정렬하여 조건에 맞게 구해주면 해결.

내가 그린 스네이크 버드

java

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.zip.InflaterInputStream;

public class Main {
	
	public static void main(String[] args) throws Exception{
		BufferedReader br = new BufferedReader((new InputStreamReader(System.in)));
		String[] split = br.readLine().split(" ");
		
		int N = Integer.parseInt(split[0]); // 과일 개수
		int L = Integer.parseInt(split[1]); // 초기 길이
		
		int[] arr = new int[N];
		split = br.readLine().split(" ");
		for(int i = 0; i < N; i++) {
			arr[i] = Integer.parseInt(split[i]);
		}
		
		Arrays.sort(arr);
		
		for(int i = 0; i < N; i++) {
			if (arr[i] <= L) {
				L += 1;
			} else {
				break;
			}
		}
		System.out.println(L);
	}
}

python

n, m = map(int,input().split())
arr = sorted(list(map(int,input().split())))
for i in arr:
  if m >= i:
    m += 1
  else:
    break
print(m)