[BaekJoon] 2028번 : 자기복제수 (Python)

코딩 1일 1문제! 오늘의 문제는 백준의 자기복제수 입니다.

2028번: 자기복제수

👨🏻‍💻 코드 ( Solution )

def is_self_replicating_number(number):
    is_replicating = False
    pow_num = pow(number, 2)
    
    number_split_idx = len(str(number)) * -1
    
    if str(number) == str(pow_num)[number_split_idx:]:
        is_replicating = True
        
    return is_replicating


def get_answer(number):
    is_replicating = is_self_replicating_number(number)
    return "YES" if is_replicating else "NO"


if __name__ == "__main__":
    for _ in range(int(input())):
        number = int(input())
        print(get_answer(number))

GitHub - SOMJANG/CODINGTEST_PRACTICE: 1일 1문제 since 2020.02.07