题目
一个 平方和三元组 (a,b,c) 指的是满足 a2 + b2 = c2 的 整数 三元组 a,b 和 c 。
给你一个整数 n ,请你返回满足 1 <= a, b, c <= n 的 平方和三元组 的数目。
示例 1:
输入:n = 5
输出:2
解释:平方和三元组为 (3,4,5) 和 (4,3,5) 。
示例 2:
输入:n = 10
输出:4
解释:平方和三元组为 (3,4,5),(4,3,5),(6,8,10) 和 (8,6,10) 。
提示:
1 <= n <= 250
来源:力扣(LeetCode)
解题思路
这个题目中三个数字最大的值的平方可能超过n,但是却不是连续的,所以我们只能在大于n的数字中枚举可能的最大平方值,并以此来判断,两个小值的平方和是否在里面。采用一般的枚举。
class Solution:def countTriples(self, n: int) -> int:temp=set([i**2 for i in range(n+1)])count=0for i in range(1,n):for j in range(1,i):if i**2+j**2 in temp:count+=2return count