[Python] 백준 1747번 - 소수&팰린드롬

문제 https://www.acmicpc.net/problem/1747

1747번: 소수&팰린드롬

 

소수 판별 함수와 팰린드롬 판별 함수만 만들면 되는 거라서 생각보다 금방 풀었다.

 

근데 자꾸 시간초과가 나서 짜증났다. 다른 사람들 코드 찾아봐도 내 코드와 비슷한 것 같아서 이유가 뭘까 했는데

어떤 사람이 소수 판별 전에 팰린드롬 판별을 먼저 해서 한 번에 여러 개를 쳐낸 걸 보았다.

그래서 나도 while문 안에서 팰린드롬 판별 먼저 하고 소수 판별을 그 다음에 하는 걸로 바꿔주었더니 시간 초과는 해결되었다!

 

잘 가다가 마지막에 99%에서 틀려서 또 한 번 짜증냈다.

보니까 1에 대한 조건을 걸어주지 않아서 틀린 것이었다.

소수 판별 함수에 n이 1인 경우 False를 리턴해주는 코드를 추가하였다.

 

끝!

 

import sys
input = sys.stdin.readline

def is_prime_number(n):
    for i in range(2, n):
        if n % i == 0:
            return False
    if n == 1:
        return False
    return True

def is_palindrome(n):
    n = str(n)
    for i in range(len(n)):
        if n[i] != n[len(n) - 1 - i]:
            return False
    return True

n = int(input())
result = n
while True:
    if is_palindrome(result):
        if is_prime_number(result):
            print(result)
            break
    result += 1

 

컴파일에러는... if문에 : 안 붙여서... ㅎㅎ