学而思编程周赛语言普及奠基组 | 2025年春第15周T3 花田-CSDN博客

本文链接：https://blog.csdn.net/guolianggsta/article/details/149028486

欢迎大家订阅我的专栏：算法题解：C++与Python实现！
本专栏旨在帮助大家从基础到进阶，逐步提升编程能力，助力信息学竞赛备战！

专栏特色
1.经典算法练习：根据信息学竞赛大纲，精心挑选经典算法题目，提供清晰的代码实现与详细指导，帮助您夯实算法基础。
2.系统化学习路径：按照算法类别和难度分级，从基础到进阶，循序渐进，帮助您全面提升编程能力与算法思维。

适合人群：

准备参加蓝桥杯、GESP、CSP-J、CSP-S等信息学竞赛的学生
希望系统学习C++/Python编程的初学者
想要提升算法与编程能力的编程爱好者

【题目描述】

小猴消灭魔物之后，来到了自己的花田中，花田被分成 $n \times m$ 个小方格，第 $i$ 行第 $j$ 列的小方格中花的种类为 $a_{i,j}$ ，小猴想要从花田中选取一个 $h \times w$ 的小花田进行实验研究，要求选出的小花田中的花的种类数要尽可能多。

由于小猴与魔物大战已经很累了，所以小猴请你帮助他找出符合要求的小花田，输出小花田中的种类数与小花田的左上角格子坐标，如果有多种答案，则优先输出行最小的，再输出列最小的。

【输入】

第一行，包含四个整数 $n, m, h, w$ 。

接下来 $n$ 行，每行包含 $m$ 个整数 $a_{i,1},a_{i,2},…,a_{i,m}$ 。

【输出】

输出包含两行：

第一行，一个整数，表示选取的小花田中花的种类数。

第二行，两个整数，表示选取的小花田的左上角格子在花田中的坐标。

【输入样例】

【输出样例】

5
1 1

【代码详解】

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

const int N = 405, M = 2000005;  // 定义常量：矩阵最大尺寸和花色最大值

int n, m, h, w;          // n: 行数, m: 列数, h: 子矩阵高度, w: 子矩阵宽度
int cnt[M], num;         // cnt: 花色计数数组, num: 当前子矩阵中不同花色数量
int a[N][N], b[N][N];    // a: 原始矩阵, b: 存储每个子矩阵的不同花色数

int main()
{
    // 输入矩阵尺寸和子矩阵尺寸
    cin >> n >> m >> h >> w;
    
    // 输入原始矩阵数据
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
        {
            cin >> a[i][j];
        }
    }

    // 计算每个h×w子矩阵的不同花色数量
    for (int i = 1; i <= n - h + 1; i++)  // 遍历所有可能的子矩阵起始行
    {
        memset(cnt, 0, sizeof(cnt));  // 重置花色计数数组
        num = 0;                       // 重置不同花色计数器
        
        // 初始化第一个子矩阵的花色统计
        for (int x = i; x <= i + h - 1; x++)
        {
            for (int y = 1; y <= w; y++)
            {
                if (cnt[a[x][y]]++ == 0)  // 如果是新出现的花色
                {
                    num++;                // 增加不同花色计数
                }
            }
        }
        b[i][1] = num;  // 存储第一个子矩阵的结果

        // 滑动窗口处理同一行的其他子矩阵
        for (int y = 2; y <= m - w + 1; y++)
        {
            // 移除最左侧一列的花色
            for (int x = i; x <= i + h - 1; x++)
            {
                if (cnt[a[x][y - 1]]-- == 1)  // 如果该花色只剩一个
                {
                    num--;                    // 减少不同花色计数
                }
            }
            
            // 添加最右侧一列的花色
            for (int x = i; x <= i + h - 1; x++)
            {
                if (cnt[a[x][y + w - 1]]++ == 0)  // 如果是新出现的花色
                {
                    num++;                       // 增加不同花色计数
                }
            }
            
            b[i][y] = num;  // 存储当前子矩阵的结果
        }
    }

    // 寻找包含最多不同花色的子矩阵
    int ans = 0, x, y;  // ans: 最大不同花色数, (x,y): 子矩阵左上角坐标
    for (int i = 1; i <= n - h + 1; i++)
    {
        for (int j = 1; j <= m - w + 1; j++)
        {
            if (b[i][j] > ans)  // 找到更大的不同花色数
            {
                ans = b[i][j];   // 更新最大值
                x = i;          // 记录位置
                y = j;
            }
        }
    }

    // 输出结果
    cout << ans << endl;        // 输出最大不同花色数
    cout << x << " " << y << endl;  // 输出子矩阵位置

    return 0;
}

【运行结果】