题目地址(633. 平方数之和)
https://leetcode-cn.com/problems/sum-of-square-numbers/
题目描述
给定一个非负整数 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
前置知识
- Java中Math函数的使用
int right = (int) Math.sqrt(c);
思路
- 可以看成是在元素为 0~target 的有序数组中查找两个数,使得这两个数的平方和为 target,如果能找到,则返回 true,表示 target 是两个整数的平方和。
- 本题和 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)。
关键点
- 先控制范围+双指针
- 注意范围上限可使用Math.sqrt©
代码
- 语言支持:Java
Java Code:
// import math;
class Solution {
public boolean judgeSquareSum(int c) {
// if(c<0) {return false;}
int left = 0;
int right = (int) Math.sqrt(c);
while (left <= right) {
int sum = left*left + right*right;
if (sum == c) {
return true;
} else if (sum > c) {
right--;
} else {
left++;
}
}
return false;
}
}
复杂度分析
令 c 为数值参数。
- 时间复杂度: O ( s q r t ( c ) ) O(sqrt(c)) O(sqrt(c))
- 空间复杂度: O ( 1 ) O(1) O(1)
大佬之作