[E排序] lc1030. 距离顺序排列矩阵单元格(排序+几何+bfs+思维)

1. 题目来源

链接：lc1030. 距离顺序排列矩阵单元格

2. 题目说明

3. 题目解析

方法一：直接排序

直接排序即可。

代码：

class Solution {
public:
    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        vector<vector<int>> res;
        for (int i = 0; i < R; ++i)     
            for (int j = 0; j < C; ++j) 
                res.push_back({i, j});

        sort(res.begin(), res.end(), [&](const vector<int> &a, const vector<int> &b) {
            return abs(a[0] - r0) + abs(a[1] - c0) < abs(b[0] - r0) + abs(b[1] - c0);
        });
        
        return res;
    }
};

方法二：桶排序

看到题解，很巧妙的一个桶排序，空间换时间。注意桶的个数选取就很厉害！

代码：

class Solution {
public:
    int dist(int r1, int c1, int r2, int c2) {
        return abs(r1 - r2) + abs(c1 - c2);
    }

    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        int maxDist = max(r0, R - 1 - r0) + max(c0, C - 1 - c0);
        vector<vector<vector<int>>> bucket(maxDist + 1);

        for (int i = 0; i < R; ++i) 
            for (int j = 0; j < C; ++j) {
                int d = dist(i, j, r0, c0);
                bucket[d].push_back({i, j});
            }
        
        vector<vector<int>> res;
        for (int i = 0; i <= maxDist; ++i) 
            for (auto e : bucket[i])
                res.push_back(e);
        
        return res;
    }
};

方法三：几何+bfs

一开始想到这种 bfs 思想按层扩展，但是边界不太会处理…四种解法：数组排序、桶排序、BFS、几何法 tql

请参考官方题解：方法三：几何法 (官方题解明显没有上面那个题解好，说的很模糊)

我们也可以直接变换枚举矩阵的顺序，直接按照曼哈顿距离遍历该矩形即可。
注意到曼哈顿距离相同的位置恰好构成一个斜着的正方形边框，因此我们可以从小到大枚举曼哈顿距离，并使用循环来直接枚举该距离对应的边框。我们每次从该正方形边框的上顶点出发，依次经过右顶点、下顶点和左顶点，最后回到上顶点。这样即可完成当前层的遍历。


他这个 while 里的方向转移不太好想…很巧妙…
四种解法：数组排序、桶排序、BFS、几何法

强烈推荐题解：

• 四种解法：数组排序、桶排序、BFS、几何法

代码：

class Solution {
public:
    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        int dx[4] = {1, 1, -1, -1}, dy[4] = {1, -1, -1, 1};       
        int maxDist = max(r0, R - 1 - r0) + max(c0, C - 1 - c0);  
        vector<vector<int>> res;
        res.push_back({r0, c0});
        int row = r0, col = c0;
        for (int dist = 1; dist <= maxDist; dist ++) {            
            row --;                             
            for (int i = 0; i < 4; ++i) {
                while ((i % 2 == 0 && row != r0) || (i % 2 != 0 && col != c0)) { // 顶点判断
                    if (row >= 0 && row < R && col >= 0 && col < C) 
                        res.push_back({row, col});
                    
                    row += dx[i], col += dy[i];
                }
            }
        }
        return res;
    }
};