글 읽기 - python 시간초과 관련 질문드립니다.

snkii35 2년 전 0

안녕하세요! 얼마전 재채점 이후로 성공에서 시간초과로 바뀌었는데, 아래 코드에서 실행 시간을 줄일 수 있는 방법이 있을까요?

import sys
input = sys.stdin.readline

def quick_sort(arr, p, q, mode):
    if not p < q:
        return
    r = make_pivot(arr, p, q, mode)
    quick_sort(arr, p, r-1, mode)
    quick_sort(arr, r+1, q, mode)
    
def make_pivot(arr, p, q, mode):
    pivot_point = p-1 
    for i in range(p, q):
        if arr[i][mode] <= arr[q][mode]:
            arr[i], arr[pivot_point+1] = arr[pivot_point+1], arr[i]
            pivot_point += 1
    arr[pivot_point+1], arr[q] = arr[q], arr[pivot_point+1]
    return pivot_point+1

n = int(input())
arr = [list(map(int, input().strip().split())) for _ in range(n)]
# sort by index 0
quick_sort(arr, 0, n-1, 0)
# sort by index 1
cur = 1
start_point = 0
while cur <= n:
    if cur == n or arr[cur-1][0] != arr[cur][0]:
        quick_sort(arr, start_point, cur-1, 1) 
        if cur == n:
            break
        start_point = cur
    cur += 1
# print sort result
for i in range(n):
    print(f'{arr[i][0]} {arr[i][1]}')

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

python 시간초과 관련 질문드립니다.

snkii35 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말