633-平方数之和

题目

633. 平方数之和 - 力扣（LeetCode） (leetcode-cn.com)

思路

题目需要尝试搜索两个数，使其之和为c, 首先我们需要确定搜索范围。要满足$a^2+b^2=c$，则必有
$$0\le a\le \sqrt{c},0\le b\le \sqrt{c},$$
因此，搜索范围限制在$[0,\sqrt{c}]$。

我们设正确结果位于$[a,b]$中，初始化$a=0,b=\sqrt{c}$，开始搜索。

如果$a^2+b^2<c$，我们可以确定正确结果的范围在$[a+1,b]$；如果$a^2+b^2>c$，我们可以确定正确结果的范围为$[a,b-1]$，直到找到正确结果则返回true，如果搜索到区间大小为0（$a>b$）则范围false。

正确性证明：

当$a^2+b^2<c$时，除了将$a$减1还可以通过将$b$加1来使得$a^2 + b^2 $的结果增大，然而，我们在之前的步骤中已经确定了正确答案位于$[a,b]$，所以无需尝试$b+1$，因此正确结果必定位于$[a+1, b]$中；对于$a^2 + b^2 > c$同理。

代码

class Solution {
public:
    bool judgeSquareSum(int c) {
        long a,b;
        a = 0;
        b = sqrt(c);

        while( a <= b) {
            long tmp = a * a + b * b;
            if (tmp == c) {
                return true;
            } else if ( tmp > c) {
                b--;
            } else {
                a++;
            }
        }return false;
    }
};