1.题目描述
给定一个非负整数 c
,你要判断是否存在两个整数 a
和 b
,使得 a2 + b2 = c
。
示例1:
输入:c = 5
输出:true
解释:1 * 1 + 2 * 2 = 5
示例2:
输入:c = 3
输出:false
2.解题思路
- 可以看成是在元素为0~target 的有序数组中查找两个数,使这两个数的平方和为target,如果能找到,则返回true,如果不能找到则返回false。
- 本题的关键是右指针的初始化,实现剪枝,从而降低时间复杂度。设右指针为 x,左指针固定为 0,为了使
的值尽可能接近 target,我们可以将 x 取为 sqrt(target)。
- 因为最多只需要遍历一次 0~sqrt(target),所以时间复杂度为 O(sqrt(target))。又因为只使用了两个额外的变量,因此空间复杂度为 O(1)。
3.代码实现
public class JudgeSqureSum {
public boolean judgeSquareSum(int target){
if(target<0){
return false;
}
int i=0,j=(int)Math.sqrt(target);
while (i<=j){
int sum=i*i+j*j;
if(sum==target){
return true;
}
else if(sum>target){
j--;
}
else {
i++;
}
}
return false;
}
public static void main(String[] args) {
int c=3;
JudgeSqureSum b=new JudgeSqureSum();
System.out.println(b.judgeSquareSum(c));
}
}
运行结果:
Leetcode 提交截图:
总体来看用双指针的效率还是不错的。