洛谷P2241 统计方形（数据加强版)

思路
数学题,直接记住结论
a,b是行数,列数
1.矩形长不等于宽
子矩形构成的矩阵的长宽是由原矩形长宽减去不同数而得
即(n-b)(m-a) (a≠b)
2.正方形长等于宽
子正方形构成的矩阵的长宽由原矩形长宽减去相同数而得
即(n-b)(m-a) (a=b)

#include <iostream>
#include <algorithm>
#include <cstdio>

using namespace std;

typedef long long LL;
const int N = 5010;

LL n, m;

int main()
{
    cin >> n >> m;
    LL res1 = 0, res2 = 0;
    for (int i = 0; i < n; i ++ )
    {
        for (int j = 0; j < m; j ++ )
        {
            if (i == j) res1 += (n - i) * (m - j);
            else res2 += (n - i) * (m - j);
        }
    }

    cout << res1 << " " << res2 << endl;

    return 0;
}