给你一个 m x n 的二进制网格 grid ,其中 1 表示某个朋友的家所处的位置。返回 最小的 总行走距离 。
总行走距离 是朋友们家到碰头地点的距离之和。
我们将使用 曼哈顿距离 来计算,其中 distance(p1, p2) = |p2.x - p1.x| + |p2.y - p1.y| 。
示例 1:
输入: grid = [[1,0,0,0,1],[0,0,0,0,0],[0,0,1,0,0]]
输出: 6
解释: 给定的三个人分别住在(0,0),(0,4) 和 (2,2):
(0,2) 是一个最佳的碰面点,其总行走距离为 2 + 2 + 2 = 6,最小,因此返回 6。
示例 2:
输入: grid = [[1,1]]
输出: 1
提示:
m == grid.length
n == grid[i].length
1 <= m, n <= 200
grid[i][j] == 0 or 1.
grid 中 至少 有两个朋友
int x_min_dist(int** grid, int n, int m) {
int l = 0, l_i = 0;
int h = m-1, h_i = n-1;
int d = 0;
while (l<h) {
while (l < m && grid[l_i][l] != 1) {
if (++l_i == n) { l_i = 0; l++; }
}
while (h > 0 && grid[h_i][h] != 1) {
if (h_i-- == 0) { h_i = n-1; h--; }
}
if (l >= h) return d;
d += h - l;
if (++l_i == n) { l_i=0; l++; }
if (h_i-- == 0) { h_i=n-1; h--; }
}
return d;
}
int y_min_dist(int** grid, int n, int m) {
int l = 0, l_j = 0;
int h = n-1, h_j = m-1;
int d = 0;
while (l<h) {
while (l < n && grid[l][l_j] != 1) {
if (++l_j == m) { l_j = 0; l++; }
}
while (h > 0 && grid[h][h_j] != 1) {
if (h_j-- == 0) { h_j = m-1; h--; }
}
if (l >= h) return d;
d += h - l;
if (++l_j == m) { l_j=0; l++; }
if (h_j-- == 0) { h_j=m-1; h--; }
}
return d;
}
int minTotalDistance(int** grid, int n, int* gridColSize){
int m = *gridColSize;
return x_min_dist(grid, n, m)+
y_min_dist(grid,n, m);
}