글 읽기 - 파이썬 시간 초과

bogsoony12 3년 전 0

여기서 왜 시간 초과가 나는지 알고 싶습니다.

N = int(input())
colorball = []
sum = 0
answer = [0 for _ in range(N)]
color = [0 for _ in range(200001)] 
for i in range(N):
    ball = list(map(int, input().split()))
    ball.append(i)
    colorball.append(ball)
colorball = sorted(colorball, key= lambda x : (x[1], x[0]))
check = [0, 0, 0, 0]
for i in colorball:
    answer[i[2]] = sum - color[i[0]]    
    sum += i[1]
    color[i[0]] += i[1]
    if check[0] == i[1] :
        answer[i[2]] -= check[0] * check[1]
        if check[2] == i[0]:
            answer[i[2]] += check[0] * check[3]
            check[3] += 1
        else :
            check[2] = i[0]
            check[3] += 1
        check[1] += 1 
    else:
        check[0] = i[1]
        check[1] = 1
        check[2] = i[0]
        check[3] = 1

for i in answer:
    print(i)

glflakcm 3년 전 0

3 15

3 5

1 15

2 10

결과:

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

문제

문제

출처

ICPC

파이썬 시간 초과

bogsoony12 3년 전 0

glflakcm 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말