【Leetcode】1925. Count Square Sum Triples

题目地址：

https://leetcode.com/problems/count-square-sum-triples/

给定一个正整数$n$，求三个数$a,b,c\le n$使得$a^2+b^2=c^2$，返回三元组$(a,b,c)$的个数。

枚举$c$，其范围是$5\sim n$，然后对撞双指针枚举$a$和$b$即可。代码如下：

public class Solution {
    public int countTriples(int n) {
        int res = 0;
        for (int i = 5; i <= n; i++) {
            for (int l = 1, r = i - 1; l <= r; ) {
                if (l * l + r * r < i * i) {
                    l++;
                } else if (l * l + r * r > i * i) {
                    r--;
                } else {
                    res += l == r ? 1 : 2;
                    l++;
                    r--;
                }
            }
        }
        
        return res;
    }
}

时间复杂度$O(n^2)$，空间$O(1)$。