蓝桥杯数正方形

在一个 N×N 的点阵上，取其中 4 个点恰好组成一个正方形的 4 个顶点，一共有多少种不同的取法？
这是一道数学归纳题, 正方形可以正着摆放, 也可以斜着摆放, 对于 n * n 的点集, 最大正方形为 (n-1)*(n-1)

考虑一个最大矩形内每个边长的正方形, 不论其在何位置

1.对于正放的正方形
对于边长为 n - 1 的正方形, 正放的有 1*1正方形 (n-1) * (n - 1) 个, 2*2正方形 (n-2) * (n - 2) 个, 3*3正方形 (n-2) * (n - 2)个, a*a正方形 (n-a)(n-a)个, (n-1) * (n - 1)正方形个
正放个数 = sum(1<=i<=n-1) ((n-i)*(n-i))

2.对于斜放的正方形

1*1 正方形中含有斜放正方形个数 0 个(这里只考虑其他小正方形没有的斜正方形, 即最大内接斜正方形) , 2*2正方形中含有斜放正方形 1 个, 3*3 有 2 个, 4*4 有3个
5*5 有4个, 我们发现 每个正方形中最大内接正方形的个数, 与正方形内部划分方格的线数相等, 所以 a*a 的正方形, 含有最大内接正方形的个数为 a - 1 个
而每一个这样的正方形在最大的 n - 1 * n - 1 中 有 (n-i)*(n-i) 个 因此

斜正方形个数 = sum(2<=i<=n-1) ((i- 1)*(n-i)*(n-i)
合并 斜+正 = sum(1<=i<=n-1) (i*(n-i)*(n-i))

#include<iostream>

using namespace std;

const int mod = 1e9 + 7;

int main() {
    int n;
    cin >> n;
    long long ans = 0;
    
    for (int i = 1; i < n; ++i) {
        ans = ans + i * (long long)(n - i) * (n - i);
        ans %= mod;
    }

    cout << ans << endl;
    return 0;
}

参考文档:Java B组蓝桥杯第十届国赛：数正方形_小小小勇的博客-CSDN博客_蓝桥杯数正方形