LeetCode866. 回文素数 Python实现

分析：题目为了节省时间，若N的位数是偶数时，那么除了11其他的都不是素数，网上有其他证明。，如 1001的位数是4，那么接下来应该从10001开始，因为在1001-9999不会有素数的（这个从10001为了节省时间），具体实现N = pow(10,N的位数）+1，注意取整！

import math
import copy


class Solution:
    def primePalindrome(self, N):
        """
        :type N: int
        :rtype: int
        """
        while True:
            if N in [2, 3, 5, 7, 11]:
                return N
            elif N == 1:
                return 2
            
            if N > 31880255: # 这个是从测试时间超时，并且题目给出的范围的最大素数是100030001
                return 100030001
            if (N > 11) and (len(str(N)) % 2 == 0):
                N = int(pow(10, len(str(N)))) + 1
                continue
            flag1 = self._isPalindrome(N)

            if flag1:
                a = math.ceil(math.sqrt(N))
                flag = self._isSushu(N, a)
                if flag:
                    return N
             
            if N % 2 != 0:
                N = N + 2
            else:
                N = N + 1

    def _isSushu(self, N, end):
        flag = False
        if N % 2 == 0:
            return flag
        for i in range(3, end + 1):
            if i % 2 != 0:
                if N % i != 0:
                    flag = True
                else:
                    flag = False
                    break
        return flag

    def _isPalindrome(self, N):
        if N % 2 == 0:
            return False
        s = list(str(N))

        s2 = copy.deepcopy(s)
        s2.reverse()
        if s == s2:
            return True
        else:
            return False


s = Solution()
import time
t1 = time.time()
print(s.primePalindrome(31880255))
print(time.time() - t1)