矩阵区域和Java
文章发布较早,内容可能过时,阅读注意甄别。
# 题目
给你一个 m x n 的矩阵 mat 和一个整数 k ,请你返回一个矩阵 answer ,其中每个 answer[i][j] 是所有满足下述条件的元素 mat[r][c] 的和:
- i - k <= r <= i + k,
- j - k <= c <= j + k 且
- (r, c) 在矩阵内。
示例 1:
输入:mat = [[1,2,3],[4,5,6],[7,8,9]], k = 1
输出:[[12,21,16],[27,45,33],[24,39,28]]
示例 2:
输入:mat = [[1,2,3],[4,5,6],[7,8,9]], k = 2
输出:[[45,45,45],[45,45,45],[45,45,45]]
提示:
- m == mat.length
- n == mat[i].length
- 1 <= m, n, k <= 100
- 1 <= mat[i][j] <= 100
# 思路
for循环
# 解法
class Solution {
public int[][] matrixBlockSum(int[][] mat, int K) {
int[][] res = new int[mat.length][mat[0].length];
for (int i = 0; i < mat.length; i++) {
for (int j = 0; j < mat[0].length; j++) {
int sum = 0;
for (int r = Math.max(i - K, 0); r <= Math.min(i + K, mat.length - 1); r++) {
for (int c = Math.max(j - K, 0); c <= Math.min(j + K, mat[0].length - 1); c++) {
sum += mat[r][c];
}
}
res[i][j] = sum;
}
}
return res;
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
# 总结
- 分析出几种情况,然后分别对各个情况实现