[BOJ 2547번] 사탕 선생 고창영(Python)

📝 문제

💡 사탕 박사 고창영은 사탕 공장을 그만 두고 초등학교 선생님으로 취직했다.
오늘은 창영이네 반 학생들과 소풍을 가는 날이다. N명의 학생은 모두 가방에 사탕을 가득 담아왔다.
그런데, 사탕을 많이 가져온 학생들이 사탕을 적게 가져온 학생들을 놀리기 시작했다.
창영이는 "얘들아~ 그러면 안되지ㅜㅜ 사탕을 모두 여기에 가져와보렴. 선생님이 공평하게 나눠줄게"
창영이는 모든 학생들이 같은 사탕을 가지게 하려고 한다.
이것이 가능한 일일까?

⚙️ 입력 : 첫째 줄에 테스트 케이스의 개수 T가 주어진다.
각 테스트 케이스는 빈 줄로 구분되어 있고, 다음과 같이 구성되어 있다.
테스트 케이스의 첫째 줄에 학생의 수 N이 주어진다.
다음 N개의 줄에는 각 학생들이 가져온 사탕의 수가 주어진다.
N은 100,000보다 작거나 같은 자연수이고, 사탕의 개수는 0보다 크거나 같은 정수이다.
학생들이 가져온 사탕 개수는 1018보다 작거나 같다.
(학생들이 가져온 사탕 개수의 합은 1018을 넘어갈 수 있다)

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

 

문제링크

2547번: 사탕 선생 고창영

---

✅ 풀이

import sys
input = sys.stdin.readline
t = int(input())
for i in range(t):
    no = input()
    n = int(input())
    sum = 0
    for j in range(n):
        sum += int(input())
    if sum % n == 0:
        print("YES")
    else:
        print("NO")

💡 입력으로 주어진 사탕의 갯수를 모두 더해서 인원수로 나누어 떨어지는지 여부만 체크하면 된다.
다만 입력 예시에 풀이에서 사용되지 않는 공백이 존재하기 때문에,
해당 공백을 저장하기 위해 더미변수 no를 별도로 생성하여 공백 1줄을 저장하도록 했다.