Leetcode 每日一题——1030. 距离顺序排列矩阵单元格

1030. 距离顺序排列矩阵单元格

给出 R 行 C 列的矩阵，其中的单元格的整数坐标为 (r, c)，满足 0 <= r < R 且 0 <= c < C。

另外，我们在该矩阵中给出了一个坐标为 (r0, c0) 的单元格。

返回矩阵中的所有单元格的坐标，并按到 (r0, c0) 的距离从最小到最大的顺序排，其中，两单元格(r1, c1) 和 (r2, c2) 之间的距离是曼哈顿距离，|r1 - r2| + |c1 - c2|。（你可以按任何满足此条件的顺序返回答案。）

这道题用暴力枚举是可以求解的，就是时间复杂度高一点，考虑到R和C都不大，可以使用桶排序的方法，将每个距离分别装桶，然后合并，具体代码如下（C++版）：

class Solution {
public:
    int distance(int r1, int c1, int r0, int c0)
    {
        return abs(r1-r0)+abs(c1-c0);
    }
    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        
        int maxLen=max(R-r0-1,r0)+max(C-c0-1,c0);
        vector<vector<vector<int>>> numMap(maxLen+1);
        for(int i=0;i<R;++i)
        {
            for(int j=0;j<C;++j)
            {
                int dis=distance(i,j,r0,c0);
                vector<int> tmp={i,j};
                numMap[dis].push_back(move(tmp));
            }
        }
        vector<vector<int>> result;
        for (int i=0;i<maxLen+1;++i)
        {
            for(auto& num : numMap[i])
            {
                result.push_back(num);
            }
        }
        return result;
    }
};

运行效果：

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/matrix-cells-in-distance-order