백준 1268번 - 임시 반장 정하기 - Python

문제 자체가 처음엔 이해가 잘 안되었는데, 다시 차근차근 읽어보니 개인별로 같은반 해본 친구들의 수를 세는 비교적 단순한 과제였다.

일단 spacebar로 구분된 input을 int로 바꾸어 school에 저장해두고, 개인별 친구수를 저장하는 student list을 만든다. 반복구문을 돌리면서 반이 겹치는 학생 두 쌍에 각각 +1 점수를 부여해간다.

최종적으로 가장 인싸만 골라내면 되니까 max로 최댓값을 알아낸 뒤에 index을 써서 몇번 학생인지 알아낸다. 동률의 경우 최솟값을 구하라고 했는데, 기본적으로 python index는 낮은 값을 리턴해주기 때문에 그대로 두면 된다. 만일 최댓값을 구하라고 했다면, max에 해당하는 index를 구해둔 뒤에 최댓값을 구하는 과정을 한 번 더 거쳐주면 된다.

# 1268

school = []
n = int(input())
student = [0]*n

for i in range(n):
    school.append(list(map(int, input().split())))

for i in range(n-1):
    temp = student[i]
    for j in range(i+1,n):
        for k in range(5):
            if school[i][k] == school[j][k]:
                student[i] += 1
                student[j] += 1
                break

print(student.index(max(student))+1)

Comments: