2020牛客寒假算法基础集训营2 E-做计数 （数学）

题目传送

题目大意：给定一个正整数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;
}