백준 10610번 30 | python

https://www.acmicpc.net/problem/10610

10610번: 30

 

문제

어느 날, 미르코는 우연히 길거리에서 양수 N을 보았다. 미르코는 30이란 수를 존경하기 때문에, 그는 길거리에서 찾은 수에 포함된 숫자들을 섞어 30의 배수가 되는 가장 큰 수를 만들고 싶어한다.

미르코를 도와 그가 만들고 싶어하는 수를 계산하는 프로그램을 작성하라.

입력

N을 입력받는다. N는 최대 105개의 숫자로 구성되어 있으며, 0으로 시작하지 않는다.

출력

미르코가 만들고 싶어하는 수가 존재한다면 그 수를 출력하라. 그 수가 존재하지 않는다면, -1을 출력하라.

---

이번 문제는 쓸데없이 어렵게 생각하려고 했던게 문제이다.

핵심만 말하자면 3의 배수를 구하는 법과 리스트에 0이 있는지만 알면 문제가 해결되는 것이었다.

입력받은 숫자를 리스트로 쪼갠 다음, z_flag와 sumval을 통해 0이 있는지와 모든 자리 수의 합이 3의 배수 인지 확인하고 모두 확인이 된다면 reverse sort를 진행한 값을 숫자로 만들어서 출력을 한다.

아래의 코드를 보자.

import sys

input = sys.stdin.readline

answer = 0
numbers = list(map(int, list(input().rstrip())))
numbers.sort(reverse=True)
n = len(numbers)

sumval, z_flag = 0, False
for number in numbers:
    if not number:
        z_flag = True
    sumval += number
    answer = answer * 10 + number

if not sumval % 3 and z_flag:
    print(answer)
else:
    print(-1)