题目大意:给定一个正整数n,问你有多少对三元组(i, j, k)满足sqrt(i) + sqrt(j) = sqrt(k)且i * j <= n。
分析:化简一下式子可以得到i + j + 2 * sqrt(i * j) = k。因此可知只要i * j是平方数即可。所以枚举一下不大于n的平方数,统计一下他们的因子数即可。
#include <iostream>
#include <vector>
using namespace std;
#define maxn 40000000+10
vector<int> p;
int main(void)
{
int n;
cin >> n;
for (int i = 1; i * i <= n; i++) p.push_back(i * i);
int ans = 0;
for (int i = 0; i < p.size(); i++) {
for (int j = 1; j * j <= p[i]; j++) {
if (p[i] % j == 0) {
if (p[i] / j == j) ans++;
else ans += 2;
}
}
}
cout << ans;
return 0;
}