[bzoj3505][数论]数三角形

Description

给定一个nxm的网格，请计算三点都在格点上的三角形共有多少个。下图为4x4的网格上的一个三角形。

注意三角形的三点不能共线。

Input

输入一行，包含两个空格分隔的正整数m和n。

Output

输出一个正整数，为所求三角形数量。

Sample Input

2 2

Sample Output

76

HINT

数据范围

1<=m,n<=1000

题解

有点妙啊…
先n++,m++
想到了寻找三点共线的方案但是只会$n^4$…
首先横竖两种方案可以很容易去掉
只考虑斜着的线段
枚举以这条边为对角线的矩阵边长i,j
那么这条线段内会经过$gcd(i,j)-1$个整点
那么这条边，实际上是可以平移的
一共可以平移$(n-i)\ast (m-j)$次
固定了两端的点 在中间选点共有$gcd(i,j)-1$种方案
一条线段的贡献就是
$2\ast (gcd(i,j)-1)\ast (n-i)\ast (m-j)$

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<ctime>
#define LL long long
#define mp(x,y) make_pair(x,y)
using namespace std;
inline int read()
{
	int f=1,x=0;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void write(int x)
{
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
}
inline void print(int x){write(x);printf(" ");}
LL C(LL n)
{
	if(n<3)return 0;
	return n*(n-1)*(n-2)/6;
}
LL gcd(LL a,LL b){return a==0?b:gcd(b%a,a);}
LL n,m;
int main()
{
	n=read();m=read();n++;m++;
	LL ans=C(n*m)-n*C(m)-m*C(n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			ans-=2*(gcd(i,j)-1)*(n-i)*(m-j);
	printf("%lld\n",ans);
	return 0;
}