题目链接: 平方数之和
有关题目
给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 a^2 + b^2 = 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 <= 2^31 - 1
题解
1、使用qsrt函数
class Solution {
public:
bool judgeSquareSum(int c) {
for (long long a = 0; a * a <= c ; a++)//a也可以是long
{
double b = sqrt(c - a * a);
if ((int)b == b)
//b = 3.1 不等于(int)b = 3
//b = 2.0 可以等于(int)b = 2
return true;
}
return false;
}
};
时间复杂度:O(sqrt( c))
空间复杂度:O(1)
2、双指针
思路 :
我们假设 a <= b,a从0开始 ,b = (long)sqrt(c)
我们定义sum = a * a + b * b 跟 c比较
分为三种情况
1.sum = c直接返回
2.sum > c 右指针减1
3.sum < c 左指针加1
class Solution {
public:
bool judgeSquareSum(int c) {
long l = 0;
long r = (long)sqrt(c);
while(l <= r)
{
if (l * l + r * r == c)//这边可能会溢出所以用sum = l * l + r * r 的值我们用long类型进行存储
return true;
else if(l * l + r * r > c)
r--;
else
l++;
}
return false;
}
};
时间复杂度:O(sqrt( c))
空间复杂度:O(1)