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

荒地电站建设

代码仅供参考，若需使用请进行一定修改，提供C++，python，java三种代码

题目

题目描述

祖国西北部有一片大片荒地，其中零星的分布着一些湖泊，保护区，矿区;整体上常年光照良好，但是也有一些地区光照不太好。某电力公司希望在这里建设多个 光伏电站，生产清洁能源对每平方公里的土地进行了发电评估，其中不能建设的区域发电量为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

思路

利用二维前缀和来计算子矩阵和

代码

#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
// 定义常量
#define inf 0x3f3f3f3f
#define int long long
// 定义数组大小
#define M 5005
// 全局变量
int n, m, r, s;
int a[M][M];

// 子矩阵的和
int add(int u, int d, int l, int r)
{
    return a[d][r] - a[d][l - 1] - a[u - 1][r] + a[u - 1][l - 1];
}

signed main()
{
    cin >> n >> m >> r >> s;
    // 输入二维数组a，同时处理前缀和
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++) {
            cin >> a[i][j];
            a[i][j] += a[i - 1][j] + a[i][j - 1] - a[i - 1][j - 1];
        }
    // 统计子矩阵的个数
    int sum = 0;
    for (int i = 1; i <= n - r + 1; i++)
        for (int j = 1; j <= m - r + 1; j++) {
            sum += (add(i, i + r - 1, j, j + r - 1) >= s);
        }
    // 输出结果
    cout << sum << endl;
    return 0;
}