【LeetCode每日一题】633. 平方数之和

原创

guangcheng0312q 2021-08-03 09:29:17 ©著作权

文章标签 链表信息熵 sms toolbar sun 文章分类 C/C++ 后端开发

©著作权归作者所有：来自51CTO博客作者guangcheng0312q的原创作品，请联系作者获取转载授权，否则将追究法律责任

【LeetCode每日一题】633. 平方数之和

今天已经坚持Day33+天了。

题目：

给定一个非负整数 c ，你要判断是否存在两个整数 a 和 b，使得 a2 + b2 = c 。

示例 1：

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

示例 2：

输入：c = 3
输出：false

预处理+查询

如果存在符合条件的a，b，那么a和b一定在1和sqrt(c)之间。

class Solution {
public:
    typedef unsigned long long ULL;
    bool judgeSquareSum(int c) {
        if (c == 0) return true;
        unordered_map<ULL, ULL> um;
        for (int i = 1; i <= sqrt(c); i++) {
            um[i*i]++;
        }

        for (int i = 1; i <= sqrt(c); i++) {
            if (i*i == c || um.count(c - i*i)) return true;
        }
        return false;
    }
};

双指针

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