LeetCode - 633. Sum of Square Numbers

LeetCode：633. Sum of Square Numbers

概述

Easy

Given a non-negative integer c, your task is to decide whether there’re two integers a and b such that a² + b² = c.

一个非负数能否是两个数的平方和。

Example

Input: 5
Output: True

Input: 2
Output: True

Input: 3
Output: False

Explanation

1 * 1 + 2 * 2 = 5

1 * 1 + 1 * 1 = 2

思路

首先，a、b两数都不可能大于$\sqrt{c}$，所以我们可以把区间缩小到[0，$\sqrt{c}$]。
声明两个下标i、j，分别从前和后遍历。
声明变量powSum=i²+j²。

1. powSum>c，令powSum变小，即j–。
2. powSum<c，令powSum变大，即i++。
3. powSum==c，返回true。
4. 否则，返回false。

由于题目没有说明a、b是否不等，所以要考虑a、b相等的情况，即while(i<=j)。

代码

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