【华为OD机试2023】荒地电站建设 python

荒地电站建设

代码仅供参考，若需使用请进行一定修改，提供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)