day7 组素数

组素数

素数就是不能再进行等分的数。比如：2 3 5 7 11 等。
9 = 3 * 3 说明它可以3等分，因而不是素数。
我们国家在1949建国。
如果只给你 1 9 4 9 这4个数字卡片，
可以随意摆放它们的先后顺序（但卡片不能倒着摆放啊，我们不是在脑筋急转弯！），
那么，你能组成多少个4位的素数呢？
比如：1949，4919 都符合要求。

解题

import itertools
import math


def isPrime(num: int):
    max = int(math.sqrt(num))+1
    for i in range(2,max):
        if num % i == 0:
            return False
    return True


lst = [1, 9, 4, 9]
count = 0
ans = []
for num in itertools.permutations(lst):
    num = eval(''.join(map(str,num)))
    if isPrime(num):
        count += 1
        ans.append(num)
print('能组成%d个4位素数，分别是：' % count)
print(ans)