【Lintcode】912. Best Meeting Point

题目地址：

https://www.lintcode.com/problem/best-meeting-point/description

给定一个二维$0-1$矩阵，要求返回一个坐标，使得所有$1$和这个坐标的曼哈顿距离和最小。返回那个曼哈顿距离。

由于曼哈顿距离是两个坐标差的绝对值之和，所以如果我们能分别最小化两个坐标上的距离，就一定能得到总体的最小曼哈顿距离。这是个几何问题。设数轴上有若干点$x_1,x_2,...,x_n$，不妨设$x_1\le x_2\le ...\le x_n$，寻找$y$使得$s(y)={\sum }_{i=1}^n|x_i-y|$最小。显然$y\in [x_1,x_n]$，否则如果$y<x_1$时有$s(y)>s(x_1)$，如果$y>x_n$时有$s(y)>s(x_n)$。我们只需考虑导数$s^{\prime }(y)$即可。如果$x_1\le ...\le x_i\le y\le x_{i+1}\le ...\le x_n$，则$s^{\prime }(y)=i-(n-i)=2i-n$，所以在$i=n/2$时$s(y)$取最小值，也就是说要取$y$为中位数即可（要根据$n$奇偶性分别讨论。这里就省略了）。代码如下：

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class Solution {
    /**
     * @param grid: a 2D grid
     * @return: the minimize travel distance
     */
    public int minTotalDistance(int[][] grid) {
        // Write your code here
        if (grid == null || grid.length == 0) {
            return 0;
        }
        
        List<Integer> row = new ArrayList<>(), col = new ArrayList<>();
        for (int i = 0; i < grid.length; i++) {
            for (int j = 0; j < grid[0].length; j++) {
                if (grid[i][j] == 1) {
                    row.add(i);
                    col.add(j);
                }
            }
        }
        
        Collections.sort(row);
        Collections.sort(col);
        
        // 取所有横坐标和纵坐标的中位数即可
        int x = row.get(row.size() / 2), y = col.get(col.size() / 2);
        
        int res = 0;
        for (int r : row) {
            res += Math.abs(r - x);
        }
        for (int c : col) {
            res += Math.abs(c - y);
        }
        
        return res;
    }
}

时间复杂度$O(rc+n\log n)$（$r$和$c$为矩阵的长宽，$n$为$1$的个数），空间$O(n)$。