力扣(Leetcode)633 平方数之和
题目描述
给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 a2 + b2 = c 。
示例 1:
输入:c = 5
输出:true
解释:1 * 1 + 2 * 2 = 5
示例 2:
输入:c = 3
输出:false
0 <= c <= 2^31 - 1
解题思路:本题和 167. Two Sum II - Input array is sorted 类似,只有一个明显区别:一个是和为 target,一个是平方和为 target。本题同样可以使用双指针得到两个数,使其平方和为 target。
本题的关键是右指针的初始化,实现剪枝,从而降低时间复杂度。设右指针为 x,左指针固定为 0,为了使 02 + x2 的值尽可能接近 target,我们可以将 x 取为 sqrt(target)。
因为最多只需要遍历一次 0~sqrt(target),所以时间复杂度为 O(sqrt(target))。又因为只使用了两个额外的变量,因此空间复杂度为 O(1)。
class Solution {
public boolean judgeSquareSum(int c) {
//利用双指针 low为低位,hight为高位 low最小为0 hight最大为c开根号
int low = 0,hight = (int)Math.sqrt(c); //Math.sqrt是double类型 这里需要强制类型转换
while(low <= hight) { //利用双指针时需要判断是否可以取等号
int sum = low*low + hight*hight; //提前先将平方和的表达式写出来,方便下边比较
if(sum == c) { //如果两数平方和等于给定的数字那么返回true
return true;
}else if(sum > c) { //如果大于我们需要让较大的数字变小,即高位向左移动,
hight--;
}else{ //最后一种情况是平方和小于给定的数字,那么需要让较小的数字右移变大,
low++;
}
}
return false; //当while循环里的不满足时,最后返回false
}
}