[백준/구현] 1268번: 임시 반장 정하기 (브론즈1, python)

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

1268번: 임시 반장 정하기

아이디어

아래 표와 같이 이차원 리스트를 만들어서 서로 같은 반이었던 적이 있으면 1, 아니면 0으로 표시해주었다.

학생/학생	1	2	3	4	5
1	1	0	0	0	0
2	0	1	0	1	0
3	0	0	1	1	1
4	0	1	1	1	1
5	0	0	1	1	1

 

전체 코드

n = int(input())

students = [[0] * n for _ in range(n)]
datas = [[int(x) for x in input().split()] for _ in range(n)]

for i in range(n):
    for j in range(5):
        for k in range(n):
            if datas[i][j] == datas[k][j]:
                students[i][k] = 1

answers = [0] * n
for idx, s in enumerate(students):
    answers[idx] = s.count(1)

def printAnswer(answers):
    for i in range(n):
        if answers[i] == max(answers):
            print(i+1)
            return

printAnswer(answers)