给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 a2 + b2 = c。
示例1:
输入: 5
输出: True
解释: 1 * 1 + 2 * 2 = 5
示例2:
输入: 3
输出: False
解题思路:
这个解题思路可以参考:两数之和 II - 输入有序数组 题目的解法:
时间复杂度为o(根号n)
空间复杂度为o(1)
代码如下:
public class JudgeSquareSum {
@Test
public void judgeSquareSumTest() {
Assert.assertTrue(judgeSquareSum(5));
Assert.assertTrue(judgeSquareSum(4));
}
public boolean judgeSquareSum(int c) {
int i = 0;
int j = c - 1;
while (i <= j) {
int var = i * i + j * j;
if (var == c) {
return true;
} else if (var < c) {
i ++;
} else {
j --;
}
}
return false;
}
}