平方根问题

问题描述

  考虑定义在两正整数上的函数SSR(平方根之和的平方):SSR(A, B) = (sqrt(A) + sqrt(B))^2。牛牛对函数值为整数的情况很感兴趣。现在给定整数n和m,请帮助牛牛计算有序对(A, B)的数量, 满足1 ≤ A ≤ n, 1 ≤ B ≤ m而且SSR(A, B)是一个整数。

解题思路

  这道题本质上是要求sqrt(AB)是整数，也就是要求AB的乘积是完全平方数，使用枚举的思路超时。
  看到一段AC的代码如下：

#include <iostream>
#include <cmath>
#include <vector>
using namespace std;

vector<bool> pend(100005, false);

void getPend(int m) {
	int rec;
	for (int i = 2, rec = i * i; rec <= m; i++, rec = i * i) {
		if (pend[rec])continue;
		for (int j = rec; j <= m; j += rec)
			pend[j] = true;
	}
}

int main() {
	int n, m;
	cin >> n >> m;
	int sn = (int)sqrt(n), sm = (int)sqrt(m); 
	long long ans = sn * sm;
	getPend(n > m ? m : n);
	for (int i = 2;; i++) {
		if (pend[i]) continue;
		if (n / i == 0 || m / i == 0) break;
		int nt = n / i, mt = m / i;
		int snt = (int)sqrt(nt), smt = (int)sqrt(mt);
		ans += snt * smt;
	}
	printf("%lld", ans);
}

还没有完全明白他的原理，先记录一下。