AcWing 1310. 数三角形 （容斥原理 + 分类讨论）

一.题目链接

二.思路

思路参考：https://www.acwing.com/solution/content/8781/
一条直线，除了两个端点，还存在的整点格点数量是gcd(i,j) - 1。i,j为该直角三角形的底边和高，直线是斜边。
证明过程

三.代码

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long ll;

ll gcd(int a, int b)  // 欧几里得算法
{
    return b ? gcd(b, a % b) : a;
}

ll C(int n)
{
    return (ll)n * (n - 1) * (n - 2) / 6;
}
int main()
{
    int n, m;
    cin >> n >> m;
    //转化为各点
    n ++, m ++;
    //答案总数：C(n*m,3) - 不成立的解
    ll res = C(n * m) - (ll)n * C(m) - (ll)m * C(n);
    for(int i = 1; i <= n; i ++)
        for(int j = 1; j <= m; j ++)
        res -= 2ll * (gcd(i, j) - 1) * (n - i) * (m - j);
    
    cout << res << endl;
    return 0;
}

四.总结

• 题型：给定n * m 的矩阵，求三角形的个数。
• 对于一个底边长为i，高为j的直角三角形，其斜边除开两个端点还含有gcd(i,j)-1个整数格点。