LeetCode刷题笔记 633.平方数之和

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-of-square-numbers

思路：

　　a^2 + b^2 = c^2，设a<b,从a=0 ,b=√c开始查找

当a^2 + b^2 > c^2时，b-1

当a^2 + b^2 < c^2时，a+1

代码：

class Solution {
public:
    bool judgeSquareSum(int c) {
        long a = 0;
        long b = (int) sqrt(c);
        while (a <= b) {
            long res = a * a + b * b;
            if (res == c) {
                return true;
            } else if ( res < c) {
                ++a;
            } else {
                --b;
            }
        }
        return false;
    }
};