给定四个整数 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(RC))
时间复杂度为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