LeetCode|633. 平方数之和

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

提示:

  • 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的类型就可以了,但是执行过程中同样会报错,不太理解为什么,在这里留个疑问吧,希望后续能够解答这个问题。

LeetCode|633. 平方数之和

 

上一篇:U-Net笔记


下一篇:【leetcode 633. 平方数之和】双指针,指针遍历方向相反