Leetcode 1925. Count Square Sum Triples [Python]

基本思路是暴力破解。但是注意sqrt给出的是float，转成int后检查下是否还是原来的数值，随后检查给出的sqrt是不是小于等于n的。

class Solution:
    def countTriples(self, n: int) -> int:
        res = 0
        for a in range(1,n):
            for b in range(1,a):
                root1 = sqrt(a**2 + b**2) 
                root2 = int(root1)
                if root1 == root2 and root2 <= n:
                    res += 2
        return res