633 平方数之和
问题描述:
给定一个非负整数 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
思路:
方法一【暴力】
经典,最容易想到的方案就是暴力了,但是暴力还要解决一些问题,比如我们遍历的起点和终点是什么,因为这里我们的c是大于等于0的,所以我们这里应该从0开始遍历,另外,c是两个平方数之和,所以我们的终点就应该是sqrt(c/2).
有了这两个边界,我们还需要解决一个问题,那就是我们怎么判断一个数是完全平方数。
一个很容易想到的思路是将这个数开方再平方,如果和原数相等,那么他就是一个完全平方数,因为sqrt方法返回的是double类型,这里需要对其进行一个强转。
Java代码:
/**
* @Description: 力扣633题题解
* @return: 返回结果
* @Author: Mr.Gao
* @Date: 2021/4/28
*/
public boolean judgeSquareSum(int c) {
int a = c/2;
a = (int)Math.sqrt(a);
for(int i = 0;i<=a;i++){
double temp = (c - Math.pow(i,2));
if((int)Math.sqrt(temp)*(int)Math.sqrt(temp)==temp){
return true;
}
}
return false;
}
- 暴力的时间总是不受眷顾的,10%
简单优化
这里可以将判断是否为全完平方数做改进。对其进行开方,如果开方完的数字等于该数字强转为(int)的话,我们就可以判断为完全平方数。
Java代码:
/**
* @Description: 力扣633题题解
* @return: 返回结果
* @Author: Mr.Gao
* @Date: 2021/4/28
*/
public boolean judgeSquareSum(int c) {
int a = c/2;
a = (int)Math.sqrt(a);
for(int i = 0;i<=a;i++){
double temp = (c - i*i);
temp = Math.sqrt(temp);
if((int)temp==temp){
return true;
}
}
return false;
}
- 这时好了点儿20%
方法二【双指针】
和上面的思路基本类似,我们通过枚举两个边界,当我们这两个边界的值的平方和大于待求数,我们就将右边界-1,当小于待求数,我们就把左边界+1,当左右边界相等的时候,我们就返回false
Java代码
public boolean judgeSquareSum(int c) {
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;
}
- 这个就很快了。96%