lintcode 697. 判断是否为平方数之和

最新推荐文章于 2024-06-03 17:05:12 发布

Sinb妃

最新推荐文章于 2024-06-03 17:05:12 发布

阅读量282

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/weixin_43981315/article/details/98480160

本文探讨了如何判断一个整数是否能表示为两个整数的平方和，通过使用集合和平方根函数，实现了高效的算法解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给一个整数 c, 你需要判断是否存在两个整数 a 和 b 使得 a^2 + b^2 = c.

样例
样例 1:

输入 : n = 5
输出 : true
说明 : 1 * 1 + 2 * 2 = 5
样例 2:

输入 : n = -5
输出 : false
输入测试数据 (每行一个参数)
如何理解测试数据？

class Solution {
public:
    /**
     * @param num: the given number
     * @return: whether whether there're two integers
     */
    bool checkSumOfSquareNumbers(int num) {
        // write your code here
        set<long> result;
        int n=sqrt(num);
        for (int i = 0; i <= n; i++) {
            result.insert(pow(i,2));
        }
        for (auto temp : result) {
            if(result.count(num-temp)){return true;break;}
        }
        return false;
    }
};