Description
给定一个 n×m 的网格,计算三点都在网格上的三角形共有多少个。
Solution
正难则反,三角形个数等于 C3(n+1)(m+1) 减去三点共线的个数。那么枚举一个矩形的长和宽,计算对角线三点共线的数量,然后乘以矩形的个数即可。
横线上和竖线上可以直接统计。
Code
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define N 1010
#define ll long long
using namespace std;
int a[N][N];
ll b[N][N];
ll calc(int x){
return (ll)x*(x-1)*(x-2)/6;
}
int gcd(int x,int y)
{
int z;
while(x%y!=0)
{
z=x%y;
x=y;
y=z;
}
return y;
}
int main()
{
int n,m;
scanf("%d %d",&n,&m);
n++,m++;
ll t=n*1ll*m,ans=calc(t);
fo(i,1,n-1)
fo(j,1,m-1)
ans-=(ll)(n-i)*(m-j)*(gcd(i,j)-1)*2;
ans-=(ll)n*calc(m)+(ll)m*calc(n);
printf("%lld",ans);
}