题目描述:
给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 a2 + b2 = c 。
示例 1:
输入:c = 5
输出:true
解释:1 * 1 + 2 * 2 = 5
示例 2:
输入:c = 3
输出:false
示例 3:
输入:c = 4
输出:true
示例 4:
输入:c = 2
输出:true
示例 5:
输入:c = 1
输出:true
提示:
0 <= c <= 231 - 1
思路:
双指针,两数相加等于一个数的平方(a2+b2=c),那么这两个数的范围都在sqrt(c)范围内,确定范围后,枚举即可。
代码:
class Solution { public: bool judgeSquareSum(int c) { long a = 0; long b = (int)sqrt(c); while (a <= b) { if (a * a + b * b == c) { return true; } else if (a * a + b * b > c) { b--; } else { a++; } } return false; } };