글 읽기 - 2243 런타임 에러 제발 도와주세요 ㅜㅜ

quf9484 2년 전 0

안녕하세요.

segtree를 재귀로 작성했는데, 런타임 에러(ImportError)가 납니다.

정답코드는 반복문, 재귀, 펜윅트리로 다양한 풀이가 있는걸 확인했습니다.

그런데 제 로직과 동일하고, 트리 크기도 크기 때문에 문제가 없다고 생각합니다.

도대체 무엇이 문제일까요...?

반나절 동안 들여다 봤는데 모르겠습니다

도와주시면 복 받으실거에요.

import sys
input = sys.stdin.readline

# index: 맛의 번호
# value: 갯수
MAX = 1000001*4
tree = [0]*MAX

# target: 수정 노드
# diff: 수정 값
# idx: target 까지 도달하는 경로에 있는 노드
def update(start, end, idx, target, diff):
    if target < start or target > end: return
    tree[idx] += diff
    if start == end: return
    mid = (start+end)//2
    update(start, mid, idx*2, target, diff)
    update(mid+1, end, idx*2+1, target, diff)
    
# n번째 순서에 있는 맛의 번호 찾기
def query(start, end, idx, target):
    if start == end:
        return start
    mid = (start+end)//2
    if target <= tree[idx*2]: # 왼 자식 보다 작거나 같으면 왼쪽으로
        return query(start, mid, idx*2, target)
    else:
        return query(mid+1, end, idx*2+1, target-tree[idx*2])

tc = int(input())
for _ in range(tc):
    cmd = list(map(int, input().rstrip().split()))
    if cmd[0] == 1:
        favor = query(1, MAX, 1, cmd[1])
        print(favor)
        update(1, MAX, 1, favor, -1) # (start,end,idx,target,diff)
    else:
        update(1, MAX, 1, cmd[1], cmd[2])

gunwookim 2년 전 1

update함수를 실행할 때 end 값에 MAX를 집어넣는데 MAX가 아니라 10^6 을 넣으면 '맞았습니다' 를 받게 됩니다.

quf9484 2년 전 0

아 그렇네요!

아직 세그트리에 대한 이해가 부족했습니다

답변 정말 감사합니다

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

문제

문제

출처

ICPC

2243 런타임 에러 제발 도와주세요 ㅜㅜ

quf9484 2년 전 0

gunwookim 2년 전 1

quf9484 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말