2021年全国高校计算机能力挑战赛C++初赛4：智能除草 农业植保无人机作为最新的设备，可以加注除草剂进行除草。每次工作可以喷洒边长为K的正方形区域。

智能除草 农业植保无人机作为最新的设备，可以加注除草剂进行除草。每次工作可以喷洒边长为K的正方形区域。现有一块边长为N的正方形农田，将其分成N*N个方格单元，已知每个单元里的杂草数量。求该植保无人机一次工作最多可以除草的数量。

题目

智能除草
农业植保无人机作为最新的设备，可以加注除草剂进行除草。每次工作可以喷洒边长为K的正方形区域。现有一块边长为N的正方形农田，将其分成N*N个方格单元，已知每个单元里的杂草数量。求该植保无人机一次工作最多可以除草的数量。

输入说明：
第一行是2个正整数，分别为N和K（1≤K≤N≤1000）。
之后N行N列正整数，表示每个单元中的杂草数量（不超过50）。
输出说明：
该植保无人机一次工作最多可以除草的数量。

输入样例：
5 2
2 2 1 1 1
1 2 1 5 6
6 1 1 4 5
2 6 1 1 1
1 1 1 1 1
输出样例：
20

按照最后一次提交的代码为评分标准。

思路

在大矩阵里遍历小矩阵，将小矩阵的和存入容器，遍历过程不断刷新小矩阵的和，然后对容器升序排序，输出容器最后一个值。

代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main()
{
	int N, n;
	cin >> N >> n;
	int arr[100][100];//如果改成arr[1000][1000],堆栈超出
	//初始化矩阵
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < N; j++) {
			cin >> arr[i][j];
		}
	}
	int sum = 0;
	vector<int> v;
	int len = N - n + 1;
	//大矩阵里遍历小矩阵
	for (int i = 0; i < len; i++) {
		for (int j = 0; j < len; j++) {
			for (int k = 0; k < n; k++) {
				for (int l = 0; l < n; l++) {
					sum += arr[i + k][j + l];
				}
			}
			//遍历的每一个小矩阵的值放入容器
			v.push_back(sum);
			sum = 0;
		}
	}
	//容器按从小到大排序
	sort(v.begin(), v.end());
	//输出容器最后一个值即最大值
	cout << v[len * len - 1];
	return 0;
}

示例