[Seop's의 코드풀이] 백준 9237 이장님의 초대 - Python

이번에 풀어볼 문제는 백준 9237번 문제은 이장님의 초대 라는 문제입니다.

문제는 다음과 같습니다

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

9237번: 이장님 초대

알고리즘 분류에는 트리로 분류되어 있지만 필자는 약간 다른 방식으로 풀이를 진행하였습니다.

 

1. 묘목들을 오름차순으로 정렬합니다.

2. 묘목의 0번째 순서는 0 을빼고 i 번째 순서에는 i 를 빼는 연산을 진행해 줍니다.

3. 전체 묘목의 수와 2번을 거친 후 묘목들 중 최댓값을 더해주면 가장 짧은 시간이 걸립니다.

4. 다 자라고 난 이후니 3번 값이 1을 더해준 값을 출력합니다. 

 

전체적인 코드는 다음과 같습니다.

질문과 피드백은 언제나 감사드립니다.

import sys
input = sys.stdin.readline

if __name__ == "__main__":
    N = int(input())
    tree = list(map(int , input().split(" ")))
    day = N
    tree = sorted(tree)
    for i in range(N) :
        tree[i] -= i
    day += max(tree) + 1
    print(day)