题目:
给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 a2 + b2 = c 。
示例 1:
输入:c = 5
输出:true
解释:1 * 1 + 2 * 2 = 5
示例 2:
输入:c = 3
输出:false
提示:
0 <= c <= 231 - 1
思路:
最重要的是转化问题,不要想着怎么处理 a^2, b^2。 想着 x + y = c,那么其实就是两数之和。
那么数组其实就是 [0, 1, 4, 9, 16, ...] 之上的两数之和
我们通过二分查找解决两数之和问题,左边开始是0, 右边开始是 sqrt(c)。
代码:
class Solution {
public:
    bool judgeSquareSum(int c) {
        int l = 0, r = sqrt(c);
        while (l <= r) {
            long long val = (long long) l * l + (long long) r * r;
            if (val == c) return true;
            else if (val > c) r--;
            else l ++;
        }
        return false;
    }
};