【CQOI2014】【BZOJ3505】数三角形

Description

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

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

Input

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

Output

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

Sample Input

2 2

Sample Output

76

数据范围

1<=m,n<=1000

HINT

Source

随随便便的组合+容斥.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 2010
#define LL long long
using namespace std;
int n,m;
LL c[MAXN*MAXN][4];
LL ans;
int gcd(int a,int b)
{
    return !b?a:gcd(b,a%b);
}
int main()
{
    scanf("%d%d",&n,&m);n++;m++;c[0][0]=1;
    for (int i=1;i<=n*m;i++)
    {
        c[i][0]=1;
        for (int j=1;j<=3;j++)  c[i][j]=c[i-1][j-1]+c[i-1][j];
    }
    ans=c[n*m][3];ans-=m*c[n][3];ans-=n*c[m][3];
    for (int i=1;i<n;i++)
        for (int j=1;j<m;j++)
        {
            int tmp=gcd(i,j)+1;
            if (tmp>=3) ans-=(LL)(tmp-2)*(n-i)*(m-j)*2;
        }
    cout<<ans<<endl;
}