백준 1356 유진수 python3

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

1356번: 유진수

 

유진수라는 친구가 있어 그냥 풀어보았습니다. (무지성 문제 선택 ㅎㅎ)

입력받은 수를 연속된 두 부분으로 나누었을 때 그 두 부분의 곱이 같으면 YES 아니면 NO를 출력합니다. 

수를 string으로 생각하고 substring의 원소들을 곱해줘서 비교하면 됩니다.

 

코드는 다음과 같습니다.

def mul(s):
    res = 1
    for c in s:
        res *= int(c)
    return res
    
n = input()

check = False
for i in range(1, len(n)):
    s1 = n[:i]
    s2 = n[i:]
    res1 = mul(s1)
    res2 = mul(s2)
    if res1 == res2:
        check = True
        break

if check == True:
    print("YES")
else:
    print("NO")