洛谷刷题 P2241

【算法1-3】暴力枚举

P2241 统计方形（数据加强版）

题目分析

本题可以使用暴力枚举，也可以使用数学公式。蜀黍菜菜，没有想到数学公式，一开始使用的是暴力枚举，后来看本题洛谷题解题解 P2241 【统计方形（数据加强版）】 - 洛谷专栏 (luogu.com.cn)发现了可用的数学公式，在此感谢这位作者。

暴力枚举思路

正方形

长方形

• 横向长方形

如下放置：

计数：

• 纵向长方形

如下放置：

计数：

暴力枚举代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;

ll n, m, min_value, max_value, square, rectangle;

int main()
{
    cin >> n >> m;
    min_value = min(m, n);
    max_value = max(m, n);
    // 正方形
    for (ll i = 1; i <= min_value; i++)
    {
        square += (min_value - i + 1) * (max_value - i + 1);
    }

    // 长方形
    for (ll i = 1; i <= min_value - 1; i++) // 纵向长方形
    {
        ll one_line = 0;
        for (ll j = i + 1; j <= min_value; j++)
        {
            one_line += min_value - (j - 1);
        }
        rectangle += one_line * (max_value - (i - 1));
    }

    for (ll i = 1; i <= min_value; i++) // 横向长方形
    {
        ll one_line = 0;
        for (ll j = i + 1; j <= max_value; j++)
        {
            one_line += max_value - (j - 1);
        }
        rectangle += one_line * (min_value - (i - 1));
    }

    cout << square << " " << rectangle;
    return 0;
}

数学公式思路

我们知道（也可能不知道哈哈哈）：矩形数量=长方形数量+正方形数量

因此使用矩形-正方形就可以求出长方形数量

正方形数量好求，因此求解与上面的暴力枚举方法一样，那么在这里就是需要知道矩形数量求解公式，如下所示：

蜀黍在这里解释一下这个公式的意思，公式用到了排列组合的思想：一个矩形是由四个顶点然后依次连线组成，那么就以左下角的顶点为(0,0)点，在横轴M+1个点中任选两个点，在纵轴N+1个点中任选两个点，这样它们就可以形成一个矩形了。如下图所示

数学公式代码

#include <bits/stdc++.h>
using namespace std;

long long n, m, min_value, max_value, square, rectangle;

int main()
{
    cin >> n >> m;
    min_value = min(m, n);
    max_value = max(m, n);
    // 正方形
    for (long long i = 1; i <= min_value; i++)
    {
        square += (min_value - i + 1) * (max_value - i + 1);
    }

    // 长方形
    rectangle = ((m + 1) * m / 2) * ((n + 1) * n / 2);
    
    cout << square << " " << rectangle - square;
    return 0;
}

总结

这道题对于对于数学功底有要求，如果不用公式，也需要有较强的逻辑梳理能力。除此之外，洛谷上面的一些测试点比较大，需要注意用到long long数据类型。感觉这道题较难