难度中等
给定一个非负整数 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 <= 2^31 - 1
双指针方法
我的答案
class Solution {
public:
bool judgeSquareSum(int c) {
long int a = 0, b = sqrt(c);
while (a <= b){
long int sum = a * a + b * b;
if (sum == c){
return true;
}else if (sum < c ){
++a;
}else{
--b;
}
}
return false;
}
};
标准答案
class Solution {
public:
bool judgeSquareSum(int c) {
long left = 0;
long right = (int)sqrt(c);
while (left <= right) {
long sum = left * left + right * right;
if (sum == c) {
return true;
} else if (sum > c) {
right--;
} else {
left++;
}
}
return false;
}
};
开始出现了错误,因为题中 c 的取值范围在 [0,2^31 - 1]。
《C++ primer plus》3.1.3节表述:int宽度可以是16位、24位或32位,甚至可以是64位,符合C++的标准。但一般计算机实现中为32位,表示-2147483648 ~ 2147483647,long为64位,上限为2^63-1。
题解将a, b 以及 sum全部定义为long类型,就可以避免错误发生。官方给出的解释是这样的。
注意:本题 cc 的取值范围在 [0,2^ - 1],因此在计算的过程中可能会发生 int 型溢出的情况,需要使用 long 型避免溢出。
这里我以为a*a和b*b超出int 上限,所以sum求和用long的类型就可以了,但是执行过程中同样会报错,不太理解为什么,在这里留个疑问吧,希望后续能够解答这个问题。