글 읽기 - 파이썬 시간초과 질문드립니다.

pepper_mis 2년 전 0

어디서 시간초과가 나는건지 잘 모르겠네요.. 어느부분이 문제인건가요...???

import sys
#input = sys.stdin.readline

MAX = 2000000
tree = [0] * (MAX+1)*4
#X추가
def tree_update(start, end, node, index): #시작범위, 끝범위, 현재위치, 입력하려고 하는 수
    if start == end: #위치에 도착하면 1 입력
        tree[node] += 1
    else:
        mid = (start+end)//2
        if mid>=index: tree_update(start, mid, node*2, index)
        else: tree_update(mid+1, end, node*2+1, index)
        tree[node] = tree[node*2] + tree[node*2+1] #노드 업데이트
    
#X번째로 작은 수 응답하고 삭제
def tree_find(start, end, node, num):
    if start == end:
        tree[node] -= 1
        return start
    
    mid = (start+end)//2
    if tree[node*2] >= num :
        result = tree_find(start, mid, node*2, num)
    else:
        result = tree_find(mid+1, end, node*2+1, num-tree[node*2])
    tree[node] = tree[node*2] + tree[node*2+1]
    return result

N = int(input())
for _ in range(N):
    T, X = map(int, input().split())
    if T == 1:
        tree_update(0, MAX, 1, X)
    else:
        print(tree_find(0, MAX, 1, X))

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

문제

문제

출처

ICPC

파이썬 시간초과 질문드립니다.

pepper_mis 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말