假设我们有四个数字 a, b, c 和 d 我们必须找到可以找到的对数 (x, y) 可以满足以下等式:x^2 + y^2 = (x*a) + ( y*b) 其中 x 在 [1, c] 范围内,y 在 [1, d] 范围内
因此,如果输入类似于 a = 2 b = 3 c = 2 d = 4,那么输出将为 1,因为一对是 (1, 1)。
让我们看看以下实现以获得更好的理解 -
def solve(a, b, c, d): ans = 0 for x in range(1,c+1): l = x*(x-a) det2 = b*b - 4*l if det2 == 0 and b%2 == 0 and 1 <= b//2 <= d: ans += 1 continue if det2 > 0: det = int(round(det2**0.5)) if det*det == det2 and (b+det) % 2 == 0: if 1 <= (b+det)//2 <= d: ans += 1 if 1 <= (b-det)//2 <= d: ans += 1 return ans a = 2 b = 3 c = 2 d = 4 print(solve(a, b, c, d))
2, 3, 2, 4输出结果
1