[파이썬] 11557번: Yangjojang of The Year

11557번: Yangjojang of The Year (acmicpc.net)

11557번: Yangjojang of The Year

가장 술을 많이 먹은 학교의 이름을 출력한다.

 

[입력]

2 <- 테스트 케이스의 개수

3 <- 입력받을 정보의 개수

Yonsei 10

Korea 1000000

Ewha 20

2 <- 입력받을 정보의 개수

Yonsei 1

Korea 1000000

 

[출력]

Korea

Korea

 

나의 풀이

t = int(input())

for i in range(t):
    school = []
    drink = []
    n = int(input())
    for j in range(n):
        a, b = input().split()
        school.append(a)
        drink.append(int(b))
    idx = drink.index(max(drink))
    print(school[idx])

'학교 숫자' 형태로 주어지는 입력을 a와 b에 나눠받는다. 

a는 school 리스트에 넣고, b는 drink 리스트에 넣는다. (b는 숫자형으로 바꿔서 넣는 것에 주의한다.)

그럼 아래처럼 된다.

school = ['Yonsei', 'Korea', 'Ewha'], drink = [10, 1000000, 20]

drink 리스트의 최댓값의 인덱스를 idx라는 변수에 저장한다.

school 리스트에 idx를 인덱스로 주어 가장 많이 술을 마신 학교의 이름을 출력한다.