给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 a2 + b2 = c 。
示例 1:
输入:c = 5
输出:true
解释:1 * 1 + 2 * 2 = 5
示例 2:
输入:c = 3
输出:false
提示:
0 <= c <= 231^{31}31 - 1
相向双指针,右指针从c\sqrt{c}c 开始,因为大于该值的数的平方已经大于c了,更别说还要加上另一个数:
cpp
class Solution {
public:
bool judgeSquareSum(int c) {
int left = 0;
int right = sqrt(c);
while (left <= right) {
if (left * left < c - right * right) {
++left;
} else if (left * left > c - right * right) {
--right;
} else {
return true;
}
}
return false;
}
};
此算法时间复杂度为O(c\sqrt{c}c ),空间复杂度为O(1)。