荒地电站建设
代码仅供参考,若需使用请进行一定修改,提供python代码
题目
题目描述
祖国西北部有一片大片荒地,其中零星的分布着一些湖泊,保护区,矿区;整体上常年光照良好,但是也有一些地区光照不太好。某电力公司希望在这里建设多个 光伏电站,生产清洁能源对每平方公里的土地进行了发电评估,其中不能建设的区域发电量为0kw,可以发电的区域根据光照,地形等给出了每平方公里年发电量x千瓦。我们希望能够找到其中集中的矩形区域建设电站,能够获得良好的收益。
输入描述
第一行输入为调研的地区长,宽,以及准备建设的电站【长宽相等,为正方形】的边长最低要求的发电量
之后每行为调研区域每平方公里的发电量
例如,输入为:
2 5 2 6
1 3 4 5 8
2 3 6 71
表示调研的区域大小为长2宽5的矩形,我们要建设的电站的边长为2,建设电站最低发电量为6
输出描述
输出为这样的区域有多少个
上述输入长宽为2的正方形满足发电量大于等于6的区域有4个。
则输出为
4
示例
示例1
输入
2 5 1 6
13 4 5 8
2 3 6 7 1
输出
3
说明
输入长为2,宽为5的场地,建设的场地为正方形场地,边长为1,要求场地的发电量大于等于6
示例2
输入
2 5 1 0
1 3 4 5 8
2 3 6 7 1
输出
10
思路
利用二维前缀和来计算子矩阵和
代码
import math
inf = math.inf
M = 5005
n, m, r, s = map(int, input().split())
a = [[0] * (m + 1) for _ in range(n + 1)]
for i in range(1, n + 1):
row = list(map(int, input().split()))
for j in range(1, m + 1):
a[i][j] = row[j - 1]
a[i][j] += a[i - 1][j] + a[i][j - 1] - a[i - 1][j - 1]
sum = 0
for i in range(1, n - r + 2):
for j in range(1, m - r + 2):
if a[i + r - 1][j + r - 1] - a[i + r - 1][j - 1] - a[i - 1][j + r - 1] + a[i - 1][j - 1] >= s:
sum += 1
print(sum)