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

思路

不失一般性，假设 a 和 b 都为非负整数。 那么 a 和 b 可能的最大值都为 ceil ( sqrt ( c ) )。
从 0 到 sqrt ( c ) 遍历 a，再判断 c - a^2 是否为平方数，如果是则返回 True。 如果遍历到最后都没有找到，那么返回 False。
时间复杂度为 O( sqrt ( c ) )

代码

class Solution {
    public boolean judgeSquareSum(int c) {
    	int upBound = (int) Math.ceil(Math.sqrt(c));
    	for(int i = 0; i <= upBound;i++) {
    		int tmp = c - i*i;
    		if(tmp >= 0 && (int)Math.sqrt(tmp) * (int)Math.sqrt(tmp) == tmp)
    			return true;
    	}
    	return false;
    	
    }
}