글 읽기 - 직접 수를 빼서 되는지 안되는지 알고싶은데 뭐가 잘못된건지 모르겠습니다.

river1345 1년 전 0

이미 많은 분들이 푸셨고, 그 방법을 소개한 글들이 많아서 어떻게 푸는지는 알고있지만,

굳이 노가다를 해서 풀어보고 싶습니다.

그래서 일일이 1를빼고 2를 빼는 방법으로 진행하려고하는데

어디에서 잘 못 된 것인지 14%에서 넘어가질 못하고 있습니다.

생각해본 반례로는

1 2 3

이있는데, 잘 통과되고 3 / 0 0 0 이나 3 / 1 2 1 등등

여러가지를 넣어봐도 잘 나오는 것 같은데 어느부분에서 막힌 것인지 도통 모르겠습니다.

예제는 잘 나옵니다.

# 완전 하나씩 빼보기
import sys

inputing = lambda : sys.stdin.readline().rstrip()
wow = lambda : map(int,inputing().split())
one = lambda : int(inputing())

import heapq
l = one()
n_list = list(wow())
n_list.sort()
while 0 in n_list:
    del n_list[n_list.index(0)]
check = "yes"
while len(n_list)>=2:
    a = heapq.heappop(n_list)
    if a%3 == 0:
        pass
    else:
        b = n_list.pop()
        a,b = sorted([a,b])
        if b < 2:
            check ="no"
            break
        else:
            a-=1
            b-=2
        if a !=0:
            heapq.heappush(n_list,a)
        if b != 0:
            heapq.heappush(n_list,b)

if check == "yes":
    if len(n_list) == 1:
        num = n_list.pop()
        print("YES" if num%3 == 0 else "NO")
    elif len(n_list) == 0:
        print("YES")
    else:
        print("NO")
else:
    print("NO")

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

문제

문제

출처

ICPC

직접 수를 빼서 되는지 안되는지 알고싶은데 뭐가 잘못된건지 모르겠습니다.

river1345 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말