BZOJ 3505: [Cqoi2014]数三角形

Description

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

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

Input

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

Output

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

Sample Input

2 2

Sample Output

76

数据范围

1<=m,n<=1000

分析

一开始我没有往容斥上面去想，然后想了个伪正解，结果发现没有考虑共线的问题。
首先有一个结论，就是在(a,b)和(x,y)连成的线段(a>x,b>y)上有gcd(a-x,b-y)+1个格点，假设以这两个端点为两个端点，第三个端点在这条线段上，那么显然有gcd(a-x,b-y)-1这么多种方案。
如果暴力的话复杂度就是O(n*n*m*m)
我们考虑把这条线段平移，使其一个端点为原点，那么与它长度相同的线段数量即为(n-i+1)*(m-i+1)，那么枚举每一个点，然后统计答案即可。
然后在坐标轴上的我们在单独讨论一下就没了。
嘲笑一波打错gcd的YMW

代码

#include <bits/stdc++.h>

#define ll long long

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

int main()
{
    int n,m;
    scanf("%d%d",&n,&m);

    int w=(n+1)*(m+1);
    ll ans=(ll)w*(w-1)*(w-2)/6;

    for (int i=2;i<=n;i++)
        ans-=(ll)(i-1)*(n-i+1)*(m+1);
    for (int i=2;i<=m;i++)
        ans-=(ll)(i-1)*(n+1)*(m-i+1);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
        {
            int s=gcd(i,j)-1;
            ans-=(ll)s*(n-i+1)*(m-j+1)*2;
        }

    printf("%lld",ans);
}