Leetcode_1030_距离顺序排序矩阵单元格_排序_bfs

11/17

初步思路

排个序就完事了

class Solution {
    public int[][] allCellsDistOrder(int R, int C, int r0, int c0) {
        int[][] ans = new int[R*C][2];
        int num=0;
        for (int i = 0; i < R; i++)
            for (int j = 0; j < C; j++) {
                ans[num][0]=i;
                ans[num++][1]=j;
            }
        Arrays.sort(ans, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return Math.abs(o1[0] - r0) + Math.abs(o1[1] - c0) - (Math.abs(o2[0] - r0) + Math.abs(o2[1] - c0));
            }
        });
        return ans;
    }
}

bfs

这个bfs可把我写吐了
在放入队列的时候，做好要用new；
因为数组传入队列的是一个地址；
如果你传入一个之前定义好的数组；
那么你修改这个数组的同时，也会对队列里的那个数组进行修改。

class Solution {
    boolean isok(int x,int y, int R, int C, boolean[][] vis) {
        if (x >= 0 && x < R && y >= 0 && y < C && !vis[x][y]) {
            return true;
        }
        return false;
    }

    public int[][] allCellsDistOrder(int R, int C, int r0, int c0) {
        int[][] ans = new int[R * C][2];
        int now = 0;
        Queue<int[]> q = new LinkedList<int[]>();
        int[][] dir = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};
        boolean[][] vis = new boolean[R][C];
        q.offer(new int []{r0,c0});
        vis[r0][c0] = true;
        int []start=new int [2];
        while (!q.isEmpty()) {
            ans[now++]=q.peek();
            start = q.remove();
            for (int i = 0; i < 4; i++) {
                int x = start[0] + dir[i][0];
                int y = start[1] + dir[i][1];
                if (isok(x,y,R,C,vis)) {
                    vis[x][y] = true;
                    q.offer(new int[]{x,y});
                }
            }
        }
        return ans;
    }
}