距离顺序排列矩阵单元格

给定四个整数 rows , cols , rCenter 和 cCenter 。有一个 rows x cols 的矩阵,你在单元格上的坐标是 (rCenter, cCenter) 。

返回矩阵中的所有单元格的坐标,并按与 (rCenter, cCenter) 的 距离 从最小到最大的顺序排。你可以按 任何 满足此条件的顺序返回答案。

单元格(r1, c1) 和 (r2, c2) 之间的距离为|r1 - r2| + |c1 - c2|。

示例 1:

输入:rows = 1, cols = 2, rCenter = 0, cCenter = 0
输出:[[0,0],[0,1]]
解释:从 (r0, c0) 到其他单元格的距离为:[0,1]

示例 2:

输入:rows = 2, cols = 2, rCenter = 0, cCenter = 1
输出:[[0,1],[0,0],[1,1],[1,0]]
解释:从 (r0, c0) 到其他单元格的距离为:[0,1,1,2]
[[0,1],[1,1],[0,0],[1,0]] 也会被视作正确答案。

示例 3:

输入:rows = 2, cols = 3, rCenter = 1, cCenter = 2
输出:[[1,2],[0,2],[1,1],[0,1],[1,0],[0,0]]
解释:从 (r0, c0) 到其他单元格的距离为:[0,1,1,2,2,3]
其他满足题目要求的答案也会被视为正确,例如 [[1,2],[1,1],[0,2],[1,0],[0,1],[0,0]]

方法一

1、利用二维数组存放矩阵中所有的坐标(时间复杂度为O(R * C))
2、创建计算距离的方法
3、利用Arrays.sort(new Comp())方法进行排序(时间复杂度为O(logR * C))

时间复杂度为O(RC * logRC)

public int[][] allCellsDistOrder(int rows, int cols, int rCenter, int cCenter) {
        int[][] arr=new int[rows*cols][];
        for (int i=0;i<rows;i++){
            for (int j=0;j<cols;j++){
                arr[i*cols+j]=new int[]{i,j};
            }
        }
        Arrays.sort(arr,(a,b)->{
            int d1=dist(a[0],a[1],rCenter,cCenter);
            int d2=dist(b[0],b[1],rCenter,cCenter);
            return Integer.compare(d1,d2);
        });
        return arr;
    }

    public static int dist(int r1,int c1,int r2,int c2){
        return Math.abs(r1-r2)+Math.abs(c1-c2);
    }

方法二

用桶排序中计数排序思想实现
1、算出最多需要几个桶(距离最大值)
2、将相同距离的数据放入相同的桶中(时间复杂度为O(RC))
3、将桶倒出,排序完成时间复杂度度(O(R
C))

时间复杂度为O(R*C)

    public int[][] allCellsDistOrder(int rows, int cols, int rCenter, int cCenter) {
        int disMax=Math.max(rCenter,rows-rCenter-1)+Math.max(cCenter,cols-cCenter-1);
        List<List<int[]>> bucket=new ArrayList<List<int[]>>();
        for (int i=0;i<=disMax;i++){
            bucket.add(new ArrayList<int[]>());
        }
        for (int i=0;i<rows;i++){
            for (int j=0;j<cols;j++){
                int d=dist(i,j,rCenter,cCenter);
                bucket.get(d).add(new int[]{i,j});
            }
        }
        int[][] arr=new int[rows*cols][];
        int index=0;
        for (int i=0;i<=disMax;i++){
            for (int[] it:bucket.get(i)){
                arr[index++]=it;
            }
        }
        return arr;
    }

方法有待改进,题目来源于力扣-1030

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值