《LeetCode之每日一题》:29.平方数之和

Posted 2021-05-16 是七喜呀！

平方数之和

• 有关题目
• • 题解

题目链接： 平方数之和

有关题目

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