双指针-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

思路:

  双指针,两数相加等于一个数的平方(a2+b2=c),那么这两个数的范围都在sqrt(c)范围内,确定范围后,枚举即可。

代码:

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

 

posted @ 2021-04-28 22:00  garigari  阅读(50)  评论(0编辑  收藏  举报