题目
给定一个非负整数 c ,你要判断是否存在两个整数 a 和 b,使得 a2 + b2 = c 。
示例 1
输入:c = 5
输出:true
解释:1 * 1 + 2 * 2 = 5
示例 2
输入:c = 3
输出:false
题解
1.预处理+查询
class Solution {
public:
typedef unsigned long long ULL;
bool judgeSquareSum(int c) {
if (c == 0) return true;
unordered_map<ULL, ULL> um;
for (int i = 1; i <= sqrt(c); i++) {
um[i*i]++;
}
for (int i = 1; i <= sqrt(c); i++) {
if (i*i == c || um.count(c - i*i)) return true;
}
return false;
}
};
2.双指针
class Solution {
public:
bool judgeSquareSum(int c) {
int j = sqrt(c);
int i = 0;
while (i <= j) {
if (i*i > c - j*j) {
j--;
} else if (i*i < c - j*j) {
i++;
} else {
return true;
}
}
return false;
}
};