[백준] 2547번: 사탕 선생 고창영 - 파이썬

• 백준 링크: 2547번: 사탕 선생 고창영
• solved.ac 난이도: 브론즈 III
• 시간 제한: 1 초
• 메모리 제한: 128 MB

문제

사탕 박사 고창영은 사탕 공장을 그만 두고 초등학교 선생님으로 취직했다.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스는 빈 줄로 구분되어 있고, 다음과 같이 구성되어 있다.

출력

각 테스트 케이스에 대해 모두에게 같은 사탕을 나눠줄 수 있으면 YES를, 없으면 NO를 출력한다.

코드

T = int(input())
for _ in range(T):
    q = input()
    N = int(input())
    li = [int(input()) for i in range(N)]
    print("YES" if sum(li)%N == 0 else "NO")