글 읽기 - 고수님들 도와주세요...(python)

sls666 4년 전 0

왜 런타임 에러가 나는지...도저히 모르겠습니다....도와주세요

import heapq

n, m = map(int, input().split())
distance = list(map(int ,input().split()))
distance.sort()

plus, minus = list(), list()
result_minus = list()
result_plus = list()

for i in range(len(distance)):
    if distance[i] <= 0:
        heapq.heappush(minus,distance[i])
    else:
        heapq.heappush(plus, (-distance[i], distance[i]))
        
while minus:
    if len(minus) > 0:
        result_minus.append(abs(heapq.heappop(minus)))
    for _ in range(m - 1):
        if minus:
            heapq.heappop(minus)
        
        
while plus:
    if len(plus) > 0:
        result_plus.append(heapq.heappop(plus)[1])
    for _ in range(m - 1):
        if plus:
            heapq.heappop(plus)
        
if result_minus:           
    max_minus = max(result_minus)
if result_plus:
    max_plus = max(result_plus)

    
final = (sum(result_minus) + sum(result_plus)) * 2

if max_minus > max_plus:
    final -= max_minus
else:
    final -= max_plus
    
print(final)

chad9349 3년 전 1

반례

1 1

-1

결과

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

문제

문제

출처

ICPC

고수님들 도와주세요...(python)

sls666 4년 전 0

chad9349 3년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말