[브론즈1]백준 1356번 유진수 - Python

1356번 유진수 - Python

사이트 바로가기 : 1356번 유진수

문제

유진수는 어떤 수를 10진수로 표현한 뒤 그 수를 두 부분으로 나눴을 때, 앞부분 자리수의 곱과 뒷부분 자리수의 곱이 같을 때를 말한다.

예를 들어, 1221은 유진수이다. 12와 21로 나눴을 때, 앞부분 자리수의 곱 1x2는 뒷부분 자리수의 곱 2x1과 같기 때문이다. 1236도 마찬가지로 유진수이다. 하지만, 1234는 아니다. 수를 나눌 때 항상 연속된 자리수를 나눠야하고, 각 부분에 적어도 한자리는 있어야 한다.

예를 들어, 12345는 총 4가지 방법으로 나눌 수 있다. 1-2345, 12-345, 123-45, 1234-5 어떤 수 N이 주어질 때, 이 수가 유진수인지 아닌지 구하는 프로그램을 작성하시오.

입력과 출력

• 입력
  첫째 줄에 수 N이 주어진다. 이 수는 2,147,483,647보다작거나 같은 자연수이다.

• 출력
  첫째 줄에 N이 유진수이면 YES, 아니면 NO를 출력한다.

풀이

from functools import reduce
N=input()
length=len(N)


check='NO'
for i in range(1,length):
    one=[int(a) for a in N[:i]]
    two=[int(b) for b in N[i:]]
    
    a=reduce(lambda x, y: x*y, one)
    b=reduce(lambda x, y: x*y, two)
    
    if a==b:
        check="YES"
        break
print(check)

마무리

import를 쓰고 싶지 않았는데 결국 써버렸다.
리스트 안에서 곱셈을 처리해주는 함수를 만들고, 이를 어떻게 적용할지 생각하는데 방법이 잘 떠오르지 않았다.