平方数之和

2024-11-04

题目:

给定一个非负整数 c ,你要判断是否存在两个整数 ab,使得 a2 + b2 = c

示例 1:

输入:c = 5
输出:true
解释:1 * 1 + 2 * 2 = 5

示例 2:

输入:c = 3
输出:false

提示:

  • 0 <= c <= 231 - 1

思路:

最重要的是转化问题,不要想着怎么处理 a^2, b^2。 想着 x + y = c,那么其实就是两数之和。

那么数组其实就是 [0, 1, 4, 9, 16, ...] 之上的两数之和

我们通过二分查找解决两数之和问题,左边开始是0, 右边开始是 sqrt(c)。

代码:

class Solution {
public:
    bool judgeSquareSum(int c) {
        int l = 0, r = sqrt(c);
        while (l <= r) {
            long long val = (long long) l * l + (long long) r * r;
            if (val == c) return true;
            else if (val > c) r--;
            else l ++;
        }
        return false;
    }
};