문제
N명의 학생들이 모인 초등학교 반에서 학급회장 선거를 하려고 한다. 그 중 3명이 회장후보로 나왔고, 이들에 대한 선호도를 N명의 학생들 각각에게 적어내도록 하였다. 세 명의 후보는 후보 1번, 후보 2번, 후보 3번이라 한다.
모든 학생은 3명의 후보 중에서 가장 선호하는 후보에게는 3점, 두 번째로 선호하는 후보에게는 2점, 가장 선호하지 않는 후보에게는 1점을 주어야 한다. 3명의 후보에 대한 한 학생의 선호 점수는 모두 다르며, 1점, 2점, 3점이 정확히 한 번씩 나타나야 한다.
후보의 최종 점수는 학생들로부터 받은 자신의 선호도 점수를 모두 더한 값이 된다. 그러면 3명의 후보 중 가장 큰 점수를 받은 후보가 회장으로 결정된다. 단, 점수가 가장 큰 후보가 여러 명인 경우에는 3점을 더 많이 받은 후보를 회장으로 결정하고, 3점을 받은 횟수가 같은 경우에는 2점을 더 많이 받은 후보를 회장으로 결정한다. 그러나 3점과 2점을 받은 횟수가 모두 동일하면, 1점을 받은 횟수도 같을 수밖에 없어 회장을 결정하지 못하게 된다.
여러분은 선호도 투표를 통해 얻은 세 후보의 점수를 계산한 후, 유일하게 회장이 결정되는 경우에는 회장으로 결정된 후보의 번호(1, 2, 3 중 한 번호)와 최고 점수를 출력하고, 회장을 결정하지 못하는 경우에는 번호 0과 최고 점수를 출력하는 프로그램을 작성하시오.
https://www.acmicpc.net/problem/2456
입력
첫째 줄에는 반의 학생들의 수 N (3 ≤ N ≤ 1,000)이 주어진다. 다음 N개의 각 줄에는 각 학생이 제출한 회장후보 3명에 대한 선호 점수가 주어지는 데, 첫 번째 점수는 후보 1번에 대한 점수이고 두 번째 점수는 후보 2번에 대한 점수이고 세 번째 점수는 후보 3번에 대한 점수이다. 이 세 점수는 서로 다르며, 1, 2, 3이 정확히 한 번씩 나타난다.
출력
학생들의 선호도 투표 결과로부터, 회장이 유일하게 결정되는 경우에는 회장으로 결정된 후보의 번호와 최고 점수를 출력하고, 유일하게 결정할 수 없는 경우에는 0과 최고 점수를 출력한다.
예제
# input
6
3 1 2
2 3 1
3 1 2
1 2 3
3 1 2
1 2 3
# output
1 13
# input
6
1 2 3
3 1 2
2 3 1
1 2 3
3 1 2
2 3 1
# output
0 12문제 풀이
import sys
input = sys.stdin.readline
N = int(input())
score = [[0] * 3 for _ in range(3)]
# 각 학생의 투표 결과 처리
for _ in range(N):
a, b, c = map(int, input().split())
score[0][a-1] += a
score[1][b-1] += b
score[2][c-1] += c
# 각 후보의 총점을 계산하는 리스트 생성
total_score = [[idx, sum(sc)] for idx, sc in enumerate(score)]
# 최고 점수 구하기
max_score = max(map(lambda x:x[1], total_score))
# 최고 점수를 받은 후보들 필터링
candidate = list(filter(lambda x: x[1] == max_score, total_score))
# 만약 최고 점수를 받은 후보가 여러 명이라면 3점을 더 많이 받은 후보를 선택
if len(candidate) >= 2:
max_three_count = max(map(lambda x: score[x[0]][2], candidate))
candidate = list(filter(lambda x: score[x[0]][2] == max_three_count, candidate))
# 만약 3점을 받은 횟수가 같은 후보가 여러 명이라면 2점을 더 많이 받은 후보를 선택
if len(candidate) >= 2:
max_two_count = max(map(lambda x: score[x[0]][1], candidate))
candidate = list(filter(lambda x: score[x[0]][1] == max_two_count, candidate))
# 여전히 후보가 여러 명이라면 회장을 결정할 수 없음
if len(candidate) >= 2:
print(0, max_score)
# 회장이 결정된 경우 후보의 인덱스 + 1(후보 번호)와 점수를 출력
else:
print(candidate[0][0]+1, candidate[0][1])파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.
