글 읽기 - (Python) 틀렸습니다...

alskdjf 2년 전 0

Python으로 제출하면 시간초과 떠서 PyPy로 제출했는데 틀렸다고 뜨네요

(누적 크기 - 같은 크기의 누적크기 - 같은 색의 누적크기 + 자신 크기) 로 구했는데

무엇이 문제일까요?

from collections import defaultdict
import sys
input = sys.stdin.readline

n = int(input())
dict_color, dict_size = defaultdict(int), defaultdict(int)
answer = [0 for _ in range(n+1)]

arr = []
for i in range(n):
    c, s = map(int,input().split())
    arr.append([c, s, i+1])

arr.sort(key = lambda x : (x[1], x[0]))

num = 0
for i in range(n):

    num += arr[i][1]

    dict_color[arr[i][0]] += arr[i][1]
    dict_size[arr[i][1]] += arr[i][1]
    
    answer[arr[i][2]] = num - dict_color[arr[i][0]] - dict_size[arr[i][1]] + arr[i][1]

    if i > 0 and arr[i-1][0] == arr[i][0] and arr[i-1][1] and arr[i][1]: answer[arr[i][2]] = answer[arr[i-1][2]]

for i in range(1, len(answer)):
    print(answer[i])

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

문제

문제

출처

ICPC

(Python) 틀렸습니다...

alskdjf 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말