荒地电站建设
代码仅供参考,若需使用请进行一定修改,提供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
思路
利用二维前缀和来计算子矩阵和
代码
import java.util.*;
public class Main {
// 定义常量
static int inf = 0x3f3f3f3f;
static int M = 5005;
// 定义全局变量
static int n, m, r, s;
static int[][] a = new int[M][M];
// 子矩阵的和
public static 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];
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
// 输入n,m,r,s
n = sc.nextInt();
m = sc.nextInt();
r = sc.nextInt();
s = sc.nextInt();
// 输入二维数组a,同时处理前缀和
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
a[i][j] = sc.nextInt();
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) ? 1 : 0;
}
}
// 输出结果
System.out.println(sum);
}
}