NumMatrix-área bidimensional y matriz de recuperación inmutable

Título

Dada una matriz bidimensional, calcule la suma de los elementos en su rango sub-rectangular. La esquina superior izquierda de la submatriz es (fila1, col1) y la esquina inferior derecha es (fila2, col2).

Inserte la descripción de la imagen aquí

En la figura anterior, la esquina superior izquierda de la submatriz (fila1, col1) = (2, 1) y la esquina inferior derecha (fila2, col2) = (4, 3), la suma de los elementos en el sub -rectángulo es 8.

Ejemplo:

Dada la matriz = [
[3, 0, 1, 4, 2],
[5, 6, 3, 2, 1],
[1, 2, 0, 1, 5],
[4, 1, 0, 1, 7 ],
[1, 0, 3, 0, 5]
]

sumRegion (2, 1, 4, 3) -> 8
sumRegion (1, 1, 2, 2) -> 11
sumRegion (1, 2, 2, 4) -> 12

Descripción:

Puede asumir que la matriz es inmutable.
El método sumRegion se llamará varias veces.
Puede asumir que row1 ≤ row2 y col1 ≤ col2.

Temas relacionados Programación dinámica

Ideas para resolver problemas

Inserte la descripción de la imagen aquí

Demostración de código

class NumMatrix {
    
    

    int[][] sums;
    public NumMatrix(int[][] matrix) {
    
    
        if(matrix.length==0)
           return;
        else {
    
    
            sums=new int[matrix.length][matrix[0].length];
            //第一个矩阵的内容大小就是matrix值
            sums[0][0]=matrix[0][0];
            //子矩阵第一行的以左上角为顶点所有值的和
            for(int i=1;i<matrix[0].length;i++)
                sums[0][i]=sums[0][i-1]+matrix[0][i];
            //同上,第一行改为第一列
            for(int i=1;i<matrix.length;i++)
                sums[i][0]=sums[i-1][0]+matrix[i][0];
            //开始计算矩阵的其他部分,以子矩阵的左上角为起点
            for(int i=1;i<matrix.length;i++)
                for(int j=1;j<matrix[0].length;j++)
                {
    
    
                    sums[i][j]=sums[i-1][j]+sums[i][j-1]-sums[i-1][j-1]+matrix[i][j];
                }
        }

    }
    
    public int sumRegion(int row1, int col1, int row2, int col2) {
    
    
        //如果从子矩阵的左上角开始计算
          if(row1==0&&col1==0)
              return sums[row2][col2];
          //起始点的横坐标是第一行
          if(row1==0)
              return sums[row2][col2]-sums[row2][col1-1];
          //起始点的纵坐标在第一列
          if(col1==0)
              return sums[row2][col2]-sums[row1-1][col2];
          //一般情况
          return sums[row2][col2]-sums[row1-1][col2]-sums[row2][col1-1]+sums[row1-1][col1-1];
    }
}

efecto

La
respuesta de información fue exitosa:
tiempo de ejecución: 15 ms, derrotando al 60.40% de los usuarios de Java
Consumo de memoria: 43.8 MB, derrotando al 92.62% de los usuarios de Java

Supongo que te gusta

Origin blog.csdn.net/tangshuai96/article/details/114278543
Recomendado
Clasificación