【2021牛客赛前集训营 提高组（第二场）】方格计数【组合数】【数学】

---

解题思路

考虑枚举两个端点，强制两个端点选，令 a 为两个端点之间 x 轴上的距离，b 为两个端点 y 轴上的距离，其实这里也可以理解为没见了一条直线（枚举一个斜率$\frac{a}{b}$)

那么这里面可以选择的点的个数有 $g=\gcd (a,b)$ 个。我们要求 $N-2$ 个小球（强制两个端点选），需要放到 g 个盒子里，相邻两个小球的盒子编号差至少为 k，方案数为
$\left(\genfrac{}{}{0ex}{}{g+1-2k-(N-3)(k-1)}{N-2}\right)$。

PS：除去断点剩下的每两个间有k-1个小球，再减去端点处的$2k-1$

最后统计方案数为：首先方案数*=2（因为相对应另一个起点会有一条斜率一样的），若x0||y0，则说明直线在坐标轴上，就不用再乘，个 $(W-x+1)\times (H-y+1)$ 就好了

---

代码

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define ll long long
using namespace std;

const int mod=1e9+7;
int T,n,w,h,d,c[2010][2010],ans;

int gcd(int x,int y) {
	if(y==0)return x;
	return gcd(y,x%y);
}

double dis(int x,int y,int xx,int yy) {
	return sqrt((xx-x)*(xx-x)+(yy-y)*(yy-y));
}

ll work(int x,int y) {
	if(x==0&&y==0)return 0;
	int g=gcd(x,y);
	int k=ceil(d/dis(0,0,x/g,y/g));
	if(k*(n-1)>g) return 0;
	int res=c[g+1-2*k-(n-3)*(k-1)][n-2];
	if(x!=0&&y!=0)res=(res+res)%mod;
	res=(ll)res*(w-x+1)*(h-y+1)%mod;
	return res;
}

int main() {
	c[0][0]=1;
	for(int i=1; i<=2000; i++)
	{
		c[i][0]=1;
		for(int j=1; j<=i; j++)
			c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
	}
	scanf("%d",&T);
	while(T--) {
		scanf("%d%d%d%d",&n,&w,&h,&d);
		if(n==1) {
			printf("%d\n",(w+1)*(h+1)%mod);
			continue;
		}
		ans=0;
		for(int i=0; i<=w; i++) 
			for(int j=0; j<=h; j++)
				ans=(ans+work(i,j))%mod;
		printf("%d\n",ans);
	}
}