LeetCode – Mejor punto de encuentro (Java)

Un grupo de dos o más personas quiere reunirse y minimizar la distancia total de viaje. Se le da una cuadrícula 2D de valores 0 o 1, donde cada 1 marca el hogar de alguien en el grupo. La distancia se calcula utilizando la Distancia de Manhattan, donde la distancia (p1, p2) = | p2.x – p1.x | + | p2.y – p1.y |.

Por ejemplo, dadas tres personas que viven en (0,0), (0,4) y (2,2):

1 - 0 - 0 - 0 - 1
|   |   |   |   |
0 - 0 - 0 - 0 - 0
|   |   |   |   |
0 - 0 - 1 - 0 - 0

El punto (0,2) es un punto de encuentro ideal, ya que la distancia total de viaje de 2 + 2 + 2 = 6 es mínima. Así que regresa 6.

Solución Java

Este problema se convierte para encontrar el valor mediano en el eje xy el eje y.

public int minTotalDistance(int[][] grid) {
    int m=grid.length;
    int n=grid[0].length;
 
    ArrayList<Integer> cols = new ArrayList<Integer>();
    ArrayList<Integer> rows = new ArrayList<Integer>();
    for(int i=0; i<m; i++){
        for(int j=0; j<n; j++){
           if(grid[i][j]==1){
               cols.add(j);
               rows.add(i);
           }
        }
    }
 
    int sum=0;
 
    for(Integer i: rows){
        sum += Math.abs(i - rows.get(rows.size()/2));    
    }
 
    Collections.sort(cols);
 
    for(Integer i: cols){
        sum+= Math.abs(i-cols.get(cols.size()/2));
    }
 
    return sum;
}