Athlete Sort

  • + 0 comments
    def set_indexer(i, arr_list):
        index_dict = {}
    
        for arr in arr_list:
            if arr[i] in index_dict.keys():
                index_dict[arr[i]].append(arr)
    
            else:
                index_dict[arr[i]] = [arr]
    
        return index_dict
    
    
    def sort_and_return(indexed_dict):
        sorted_list = sorted(indexed_dict.keys())
        sorted_arr = []
        for key in sorted_list:
            sorted_arr.extend(indexed_dict[key])
    
        return sorted_arr
    
    
    
    if __name__ == '__main__':
        nm = input().split()
    
        n = int(nm[0])
    
        m = int(nm[1])
    
        arr = []
    
        for _ in range(n):
            arr.append(list(map(int, input().rstrip().split())))
    
        k = int(input())
        
        f_arr = sort_and_return(set_indexer(k, arr))
        
        for arr in f_arr:
            for _ in arr:
                print(_, end=' ')
            print()