2020 蓝桥杯大学 B 组省赛模拟赛（一）E. 结果填空：方阵

•  1000ms
•  131072K

广场上的小朋友们排成了整齐的方阵。具体来说，我们可以把每个小朋友看做是一个点，那么小朋友们就形成了 n\times nn×n 的点阵。方阵中，小朋友 AA 和小朋友 BB 互相可以看见，当且仅当二人之间的连线不经过别的小朋友，且他们之间的距离不超过 kk （因为太远就看不见了）。我们想知道有多少对小朋友互相可以看见。(A,B)(A,B) 与 (B,A)(B,A) 算同一对。

例如， n=2,k=1n=2,k=1 时答案为 44， n=2,k=2n=2,k=2 时答案为 66（距离为 11 的有 44 对，距离为 \sqrt 22​ 的有 22 对）， n=3,k=2n=3,k=2 时答案为 2020 。

现在我们想要知道，当 n=1000,k=500n=1000,k=500 时的答案是多少。由于答案过大，请回答对 10^9+7109+7 取模后的结果。

思路：设 A（

x1,y1），B（x2,y2）；x=x1-x2，y=y1-y2;

如果gcd(x, y)  == 1 即则AB可以互相看见，否则存在x = x / gcd,y = y / gcd的点阻挡AB.

如果x 或 y == 0,即AB相邻，总个数为2*(n-1)*n;

否则寻找图内边长为x和y的矩形的个数，每个矩形内有两对，共有2*(n-x)*(n-y)对。

#include <bits/stdc++.h>
using namespace std;

int main()
{
	int n, k, res;
	cin >> n >> k;
	res = 2 * n * (n - 1) % 1000000007;
	for (int i = 1; i  < n; ++i){
		for (int j = 1; j < n; ++j){
			if (__gcd(i, j) == 1 && i * i + j * j <= k * k){
				res += (n - i) * (n - j) * 2;
				res %= 1000000007;
			}
		}
	}
	cout << res << endl;
	return 0;
}