LeetCode633. 平方数之和(数学,三角形特征)

https://leetcode-cn.com/problems/sum-of-square-numbers

给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 aa + bb = c。
示例1:
输入: 5
输出: True
解释: 1 * 1 + 2 * 2 = 5

示例2:
输入: 3
输出: False

/*思路
根据三角形三条边的特性
a为短边,b为长边, 若a*a + b*b > c*c 则长边减一,否则短边减一
*/

bool judgeSquareSum(int c){
    long long val = sqrt(c);
    long long a = 0;
    long long b = val;
        
    while( b>=a ){
        long long sum = a*a + b*b;
        if(c == sum) return true;
        if(c < sum) b--;
        if(c > sum) a++;
    }
    
    return false;
}

猜你喜欢

转载自blog.csdn.net/qq_34595352/article/details/93202536