글 읽기 - Python3 시간제한을 늘려주시면 감사하겠습니다.

h0ngjun7 7년 전 0

O((N+K) log (N+K))로 작성하였는데 시간초과가 발생합니다.

제 로컬에서는 최대 데이터에 대해 5~6초 정도에 답을 내놓는데, 채점 환경에서는 그렇지 않은 것 같습니다.

Python3 시간제한을 늘려주시면 감사하겠습니다.

마지막으로 제출한 제 코드를 아래에 첨부해드립니다.

import sys
import queue
#ifp = sys.stdin
#ofp = sys.stdout
#sys.stdin = open("input.txt", "r")
#sys.stdout = open('output.txt', 'w')
N, K = map(int, input().split(' '))
A = []
for i in range(N):
	x, y = map(int, input().split(' '))
	A.append([x, y])
for i in range(K):
	A.append([int(input()), 2000000])
ans = 0
pq = queue.PriorityQueue(N)
for x in A:
	if x[1] != 2000000:
		pq.put(-x[1])
	else:
		if (not(pq.empty())):
			t = -pq.get()
			ans += t
print (ans)
#sys.stdin = ifp
#sys.stdout = ofp

h0ngjun7 7년 전 1

시간 제한 안늘려주셔도 될 것 같습니다.

아래와 같이 입력을 좀 더 빠르게 고쳤더니 4.4초로 맞다고 판정되네요.

import queue
from sys import stdin, stdout
N, K = stdin.readline().split()
N = int(N)
K = int(K)
A = []
for i in range(N):
	x, y = stdin.readline().split()
	x = int(x)
	y = int(y)
	A.append([x, y])
for i in range(K):
	A.append([int(stdin.readline()), 2000000])
A.sort()
ans = 0
pq = queue.PriorityQueue(N)
for x in A:
	if x[1] != 2000000:
		pq.put(-x[1])
	else:
		if (not(pq.empty())):
			t = -pq.get()
			ans += t
print (ans)

sekeipost 2년 전 0

혹시 input 데이터에 대한 시간확인은 어떤식으로 하는건가요..? input 예시가 따로 있는것도 아닌데, 궁금합니다 ㅠ

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

문제

문제

출처

ICPC

Python3 시간제한을 늘려주시면 감사하겠습니다.

h0ngjun7 7년 전 0

h0ngjun7 7년 전 1

sekeipost 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말