글 읽기 - (Python) heap sort 시간초과

wockd9600 3년 전 0

pypy에서는 어떻게든 맞았는데 python에서는 여전히 시간초과가 떠서 질문드립니다.

어디가 문제인가요??

import sys

nums = int(input())
arr=[]
temp = 0
for i in range(nums):
    arr.append(int(sys.stdin.readline()))


def heap(arr):
    for i in range(1, len(arr)):
        c = i
        while True:
            root = int((c-1)/2)
            if arr[c] > arr[root]:
                temp=arr[c]
                arr[c]=arr[root]
                arr[root]=temp
            c = root
            if c<=0:
                break

def heapify(arr, k):
    root=0
    while True:
        c=root*2+1
        if(c>k):
            break
        if arr[c]<arr[c+1] and c<=k-1:
            c+=1
        if arr[c] > arr[root]:
            temp=arr[c]
            arr[c]=arr[root]
            arr[root]=temp
        root=c
        
        


def heapsort(arr):
    k = len(arr)-1
    while True:

        temp=arr[0]
        arr[0]=arr[k]
        arr[k]=temp

        k-=1
        heapify(arr, k)
        if k<=0:
            return
        
heap(arr)
heapsort(arr)


for i in range(nums):
    print(arr[i])

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

문제

문제

출처

ICPC

(Python) heap sort 시간초과

wockd9600 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말