Question

目前试图找到所有原始的pytagorean三元组，直到某个数n，使用公式a = m ^ 2-n ^ 2，b = 2mn，c = m ^ 2 + n ^ 2。这是代码：

    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。 Theres 16，c低于100，但我们只得到6.干杯

Answer 1

除了第一个条件，m＆amp; 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)

Python原始Pythagorean三重代码无法正常工作

1 个答案: