글 읽기 - 반례들은 다 맞는데 어디가 잘못된걸까요?

maecsan1 3년 전 0

python3으로 하면 시간초과가 나고 pypy3으로 하면 틀렸습니다가 나옵니다.

반례들을 다 집어 넣어봤을 때는 정답이 출력되는데 왜그럴까요? 도와주세요 ㅠ

import sys
input = sys.stdin.readline

N = int(input())
nums = list(map(int, input().split()))
nums.sort()

start = 0
end = N-1
mid = N//2
hap = nums[start] + nums[end] + nums[mid]
ans = [nums[start], nums[mid], nums[end]]

def ab(num):
    if num < 0: return -num
    return num

def sol(s, e, m):
    global start, end, mid, hap, ans
    if s == m or e == m or hap == 0: return
    start = s
    end = e
    mid = m
    while True:
        if hap > 0:
            if (end-1) <= mid: mid -= 1
            else: end -= 1
        elif hap < 0:
            if (start+1) >= mid: mid += 1
            else: start += 1
        else:
            ans = [nums[start], nums[mid], nums[end]]
            break
        if mid == start or mid == end: break
       
        tmp = nums[start] + nums[end] + nums[mid]

        if ab(tmp) < ab(hap):
            hap = tmp
            ans = [nums[start], nums[mid], nums[end]]

for i in range(N//2):
    start = i
    end = (N-1)-i
    sol(start, end, mid)
    mid = (N-1)//2
    
for i in ans:
    print(i, end = ' ')

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

문제

문제

출처

ICPC

반례들은 다 맞는데 어디가 잘못된걸까요?

maecsan1 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말