633. 平方数之和 中等
问题描述
给定一个非负整数 c
,你要判断是否存在两个整数 a
和 b
,使得 a2 + b2 = c
。
示例 1:
输入:c = 5
输出:true
解释:1 * 1 + 2 * 2 = 5
示例 2:
输入:c = 3
输出:false
提示:
0 <= c <= 231 - 1
解题思路
基于双指针的思想,
代码实现
public static boolean judgeSquareSum(int c) {
long i=0; //左指针
long j= (int) Math.sqrt(c); //右指针
while (i<=j){
long total=i*i+j*j;
if (total == c){
return true;
}else if (total > c) { //收缩右指针
j--;
}else { //收缩左指针
i++;
}
}
return false;
}
踩坑点
这题有个比较坑的测试例:c=2147483600
为了避免溢出而导致的Wrong Answer,我们选择了long型数据来接受i,j,total变量而非int型数据。