[BOJ/백준] 2547 - 사탕 선생 고창영 (Python)

문제

문제링크

학생들이 가져온 사탕을 모두 모아 남기지 않고  모든 학생들에게 공평하게 나누어줄 수 있는가?

 

풀이

각 테스트 케이스는 빈 줄로 구분되어 있고 각 테스트 케이스 첫 줄에 학생 수가 입력되므로

empty와 n에 각각을 입력받은 후, 사탕의 수를 리스트에 저장한다.

사탕 수의 합을 구하고 학생 수로 나눈 나머지가 0이라면 공평히 나누어줄 수 있는 것이다.

 

코드

t=int(input())
for _ in range(t):
    empty=input()
    n=int(input())
    candy=[int(input()) for _ in range(n)]
    if sum(candy)%n==0:
        print('YES')
    else:
        print('NO')