pta题库答案python 素数排位

pta题库答案Python素数排位是指解决某些在线编程挑战（如Pascal's Triangle of Primes）中关于素数排列的问题。在这个问题中，你需要生成帕斯卡三角形（一种数学图形），其中每个数字要么是一个素数，要么是它上方两个素数的乘积。例如，对于给定的高度，你需要计算并显示包含适当素数的帕斯卡三角形。

Python中实现这个功能的一个常见做法是使用`itertools.combinations()`函数生成可能的组合，并通过检查每个数是否为素数来填充三角形。你可以利用Python的标准库函数如`isprime()`来自`math`模块来判断一个数是否为素数。

下面是一个简化的示例代码片段：

import itertools
from math import isqrt

def is_prime(n):
    if n <= 1:
        return False
    for i in range(2, isqrt(n) + 1):
        if n % i == 0:
            return False
    return True

def prime_pascals_triangle(height):
    triangle = [[1]]
    primes = [2]  # 初始化第一个素数位置
    for _ in range(1, height):
        new_row = []
        for combination in itertools.permutations(primes, r=2):  # 取前两个素数组合
            product = combination[0] * combination[1]
            if is_prime(product):
                new_row.append(product)
        primes.extend(new_row)  # 更新当前行后的素数列表
        triangle.append(new_row)
    return triangle

height = int(input("请输入帕斯卡三角形的高度："))
result = prime_pascals_triangle(height)
for row in result:
    print(row)