<Hackerrank 문제풀이: 파이썬> Athlete Sort

 

Medium 문제로 분류되어 있지만, 생각보다 매우 간단한 문제였다:

#!/bin/python3

import math
import os
import random
import re
import sys



if __name__ == '__main__':
    nm = input().split()

    n = int(nm[0])
    m = int(nm[1])

    arr = []

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

    k = int(input())
    new_arr = sorted(arr, key=lambda x: x[k])
    for j in new_arr:
        for k in range(len(j)):
            j[k] = str(j[k])
        print(' '.join(j))
    

sorted의 lambda를 사용하여, 입력된 k 필드의 순위에 따라 순서를 정하고 이를 문자열로 변환 후, 각각 출력해주는 방식을 선택했다.