[LeetCode/Medium] 1314. Matrix Block Sum

1. Input

1) m x n int 행렬 mat

2) int 변수 k

 

 

2. Output

1) int 행렬 answer

2) answer[i][j] == 모든 mat[r][c]값의 합 

3) i - k <= r <= i + k,

4) j - k <= c <= j + k, and

5) (r, c)는 mat에 있는 유효한 위치임

  - 3 x 3 행렬 mat, k==2일 때 answer[0][0]을 구한다고 하면 -2 <= r <= 2이지만, 실제로는 0 <= r <= 2로 계산

 

3. Constraint

1) m == mat.length

2) n == mat[i].length

3) 1 <= m, n, k <= 100

4) 1 <= mat[i][j] <= 100

 

4. Example

Input: mat = [[1,2,3],[4,5,6],[7,8,9]], k = 1
Output: [[12,21,16],[27,45,33],[24,39,28]]

설명:

  - answer[0][0]: -1 <= r <= 1, -1 <= c <= 1이므로 mat[0][0] + mat[0][1] + mat[1][0] + mat[1][1] = 1+2+4+5 = 12

  - answer[1][0]: 0 <= r <= 2, -1 <= c <= 1이므로 mat[0][0] + mat[0][1] +  mat[1][0] + mat[1][1] + mat[2][0] + mat[2][1] = 1+2+4+5+7+8 = 27

 

5. Code

1) 첫 코드(2022/08/22)

int m = mat.length;
int n = mat[0].length;
int[][] answer = new int[m][n];
for(int i=0 ; i<m ; i++){
    for(int j=0 ; j<n ; j++){
        int r1 = i-k>=0 ? i-k : 0;
        int r2 = i+k<m ? i+k : m-1;
        int c1 = j-k>=0 ? j-k : 0;
        int c2 = j+k<n ? j+k : n-1;
        int sum = 0;
        for(int r=r1 ; r<=r2 ; r++)
            for(int c=c1 ; c<=c2 ; c++)
                sum += mat[r][c];
        answer[i][j] = sum;
    } // for j
} // for i
return answer;

  - 유효 범위를 만들기 위해 r1, r2, c1, c2 사용