华为OD机试真题-区域发电量统计【2023Q1】

题目描述

祖国西北部有一片大片荒地，其中零星的分布着一些湖泊，保护区，矿区.整体上常年光照良好，但是也有一些地区光照不太好。某电力公司希望在这里建设多个光伏电站，生产清洁能源对每平方公里的土地进行了发电评估，其中不能建设的区域发电量为0kw，可以发电的区域根据光照，地形等给出了每平方公里年发电量x千瓦。我们希望能够找到其中集中的矩形区域建设电站，能够获得良好的收益.
输入描述
第一行输入为调研的地区长，宽，以及准备建设的电站【长宽相等，为正方形】的边长最低要求的发电量
之后每行为调研区域每平方公里的发电量 输出描述 输出为这样的区域有多少个
示例1：
输入
2 5 2 6
1 3 4 5 8
2 3 6 7 1
输出
4
输入说明
调研的区域大小为长2宽5的矩形，我们要建设的电站的边长为2，建设电站最低发电量为6.
输出说明
长宽为2的正方形满足发电量大于等于6的区域有4个

示例2：

输入

2 5 1 6
1 3 4 5 8
2 3 6 7 1
输出 3

答题思路

首先计算输入矩阵的前缀和矩阵，然后遍历前缀和矩阵，计算边长为c的正方形子矩阵的发电量和。如果这个和大于等于最低发电量要求k，则满足条件的区域数量加一。