题目:
给定一个非负整数 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;
}
};