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

最新推荐文章于 2024-11-04 11:37:33 发布

原创最新推荐文章于 2024-11-04 11:37:33 发布 · 279 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

力扣专栏收录该内容

47 篇文章

订阅专栏

这篇博客介绍了如何使用双指针法解决寻找两个整数平方和等于给定非负整数的问题。文章提供了C++和Python两种语言的实现，并分析了为何这种方法适用。示例展示了不同输入值下的解决方案，包括效率更高的C++代码实现。

难度中等

给定一个非负整数 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