2

目前正在尝试使用公式 a = m^2-n^2, b = 2mn, c = m^2 + n^2 找到所有原始毕达哥拉斯三元组直到某个数字 n。这是代码：

    def prim(k):
        primlist=[]
        for m in range(1,k):
            for n in range(m+1,k+1):
                if m**2+n**2>k:
                    break
                if m%2==1 and n%2==1:
                    break
               if n**2-m**2==0 or 2*m*n ==0 or n**2+m**2==0:
                    break

                if gcd(m,n)!=1:
                    break


                primlist.append([n**2-m**2,2*m*n,n**2+m**2])
        return primlist
    print(prim(100))

我们希望找到所有的毕达哥拉斯三元组，但是有些缺失，例如 20,21,29。有 16 个 c 低于 100，但我们只得到 6 个。干杯

4

1 回答 1

2

除了第一个条件，其中m&n太大，你不想跳出内部循环，只需继续下一次迭代。

以下测试完全是多余的，因为您的测试m永远n不会是真的。

if n**2-m**2==0 or 2*m*n ==0 or n**2+m**2==0:

所以我已经从代码中删除了它们。你没有提供一个gcd函数，或者从标准库中导入一个函数，所以我提供了我自己的。

def gcd(a, b):
    '''  Greatest common divisor of a & b '''
    while b:
        a, b = b, a % b
    return a

def prim(k):
    primlist=[]
    for m in range(1,k):
        for n in range(m+1,k+1):
            if m**2 + n**2 > k:
                break
            if m%2==1 and n%2==1:
                continue
            if gcd(m,n)!=1:
                continue

            primlist.append([n**2 - m**2, 2*m*n, n**2 + m**2])
    return primlist

print(prim(100))

输出

[[3, 4, 5], [15, 8, 17], [35, 12, 37], [63, 16, 65], [5, 12, 13], [21, 20, 29], [45, 28, 53], [77, 36, 85], [7, 24, 25], [55, 48, 73], [9, 40, 41], [33, 56, 65], [65, 72, 97], [11, 60, 61], [39, 80, 89], [13, 84, 85]]

FWIW，这是编写该代码的一种更有效（并且更 Pythonic）的方式。我们制作了一个生成器，而不是构建一个列表。这样我们就可以打印或使用三元组，当然我们可以很容易地将它们收集到一个列表中，例如list(prim(100)).

def gcd(a, b):
    '''  Greatest common divisor of a & b '''
    while b:
        a, b = b, a % b
    return a

def prim(k):
    for m in range(1, k):
        for n in range(m+1, k+1):
            m2, n2 = m * m, n * n
            if m2 + n2 > k:
                break
            if m % 2 and n % 2:
                continue
            if gcd(m, n) > 1:
                continue

            yield n2 - m2, 2*m*n, n2 + m2

for i, t in enumerate(prim(100), 1):
    print(i, t)

输出

1 (3, 4, 5)
2 (15, 8, 17)
3 (35, 12, 37)
4 (63, 16, 65)
5 (5, 12, 13)
6 (21, 20, 29)
7 (45, 28, 53)
8 (77, 36, 85)
9 (7, 24, 25)
10 (55, 48, 73)
11 (9, 40, 41)
12 (33, 56, 65)
13 (65, 72, 97)
14 (11, 60, 61)
15 (39, 80, 89)
16 (13, 84, 85)
于 2017-12-07T14:37:49.957 回答