백준 12788 제 2회 IUPC는 잘 개최될 수 있을까? (파이썬)

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

12788번: 제 2회 IUPC는 잘 개최될 수 있을까?

필요한 팬의 수는 팀원의 수 * 팀원 당 인원으로 정해집니다. (M * K)

주어진 리스트를 내림차순으로 정렬한 뒤,

팬을 다 빌려줄 수 있는 상황이 될 때까지 (pen < 0) 연필 수를 빼는 작업을 반복합니다.

만약 줄 수 있는 팬을 다 준 상황인데도 필요한 팬이 남아있다면 그땐 stress 처리를 합니다.

 

 

N = int(input())
M, K = map(int, input().split())
A = sorted(list(map(int, input().split())), reverse=True)
pen = M * K
cnt = 0
for p in A:
    if pen > 0:
        cnt += 1
        pen -= p
    else:
        break
if pen > 0:
    print('STRESS')
else:
    print(cnt)