【双指针】633.平方数之和

目录

枚举

更好的写法 c++

双指针

---

 

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 <= 231 - 1

 

枚举

C++

class Solution {
public:
    bool judgeSquareSum(int c) {
        for (int i=0;i<=sqrt(c);i++)
        {
            int b= int (sqrt(c-i*i));
            if (i*i+b*b == c)  return true;
        }
        return false;
    }
};

更好的写法 c++

class Solution {
public:
    bool judgeSquareSum(int c) {
        for (long a = 0; a * a <= c; a++) {
            double b = sqrt(c - a * a);
            if (b == (int)b) {
                return true;
            }
        }
        return false;
    }
};

python

 

双指针

为什么这题可以用双指针？

https://leetcode-cn.com/problems/sum-of-square-numbers/solution/shuang-zhi-zhen-de-ben-zhi-er-wei-ju-zhe-ebn3/

c++

public:
    bool judgeSquareSum(int c) {
    
        long low = 0,high = sqrt(c);

        while(low<=high)
        {
           long sum = low*low + high*high;

            if(sum==c)
            {
                return true;
            }
            else if(sum<c)
            {
              ++low;
            } 
            else
            {
                --high;
            }
        }
        return false;
    }
};

python

class Solution:
    def judgeSquareSum(self, c: int) -> bool:
        low, high = 0, int(c**0.5)
        while low<=high:
            sumOf = low*low+high*high
            if sumOf==c: return True
            elif sumOf<c: low += 1
            else: high -= 1
        return False