leetcode 396 最佳碰头地点

问题描述:

有一队人(两人或以上)想要在一个地方碰面,他们希望能够最小化他们的总行走距离。给你一个 2D 网格,其中各个格子内的值要么是 0,要么是 1。 1表示某个人的家所处的位置。


原因分析:

public class 最佳碰头地点 {

    public static int minTotalDistance(int[][] grid){
        int N = grid.length;
        int M = grid[0].length;
        int[] iOnes = new int[N];//统计每一行有多少1
        int[] jOnes = new int[M];//统计每一列有多少1
        for(int i =0;i<N;i++){
            for(int j =0;j<M;j++){
                if(grid[i][j] == 1){
                    iOnes[i]++;
                    jOnes[j]++;
                }
            }
        }
        //找到最少行走距离所在行
        int total = 0;
        int i = 0;
        int j = N-1;
        int iRest = 0;
        int jRest = 0;
        while(i < j){
            if(iOnes[i] + iRest <=iOnes[j] + jRest){
                total += iOnes[i] + iRest;
                iRest += iOnes[i++];
            }else{
                total += iOnes[j] + jRest;
                jRest += iOnes[j--];
            }
        }


        //找到最少行走距离所在列
        i = 0;
        j = M-1;
        iRest = 0;
        jRest = 0;
        while(i < j){
            if(jOnes[i] + iRest <= jOnes[j] + jRest){
                total += jOnes[i] + iRest;
                iRest +=jOnes[i++];
            }else{
                total += jOnes[j] + jRest;
                jRest +=jOnes[j--];
            }
        }

        return total;
    }
}


版权声明:本文为qq_41939120原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。