LeetCode题解2020/11/17 距离顺序排列矩阵单元格

最新推荐文章于 2022-11-13 15:30:29 发布

原创最新推荐文章于 2022-11-13 15:30:29 发布 · 146 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode题解专栏收录该内容

5 篇文章

订阅专栏

博客给出 R 行 C 列矩阵及其中一坐标 (r0, c0)，要求返回矩阵所有单元格坐标，并按到 (r0, c0) 的曼哈顿距离从小到大排序。作者思路是用 BFS 以给定点为起始点向周围发散，用标记矩阵避免重复扫描，将出队列的坐标点存储。

每日一题

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

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

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

思路：我的思路是BFS（广度优先搜索），以给定点为起始点向周围（上下左右）以一个单位距离发散（曼哈顿距离），用一个标记矩阵来判断是否存在重复扫描，若已经扫描过了则不重复进入队列，出队列之后即就是按照曼哈顿距离大小排列得坐标点，将其存储起来即可，难度不大。

class Solution {
public:
    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        vector<vector<int>> loc(R,vector<int>(C,0));
        vector<vector<int>> ans;
        queue<int> que_x;
        queue<int> que_y;

        loc[r0][c0]=1;
        que_x.push(r0);
        que_y.push(c0);
        while(!que_x.empty()){
            int temp_x=que_x.front();
            int temp_y=que_y.front();
            que_x.pop();
            que_y.pop();
            if(temp_x-1>=0&&loc[temp_x-1][temp_y]==0){
                loc[temp_x-1][temp_y]=1;
                que_x.push(temp_x-1);
                que_y.push(temp_y);
            }
            if(temp_y-1>=0&&loc[temp_x][temp_y-1]==0){
                loc[temp_x][temp_y-1]=1;
                que_x.push(temp_x);
                que_y.push(temp_y-1);
            }
            if(temp_y+1<C&&loc[temp_x][temp_y+1]==0){
                loc[temp_x][temp_y+1]=1;
                que_x.push(temp_x);
                que_y.push(temp_y+1);
            }
            if(temp_x+1<R&&loc[temp_x+1][temp_y]==0){
                loc[temp_x+1][temp_y]=1;
                que_x.push(temp_x+1);
                que_y.push(temp_y);
            }
            vector<int> temp;
            temp.push_back(temp_x);
            temp.push_back(temp_y);
            ans.push_back(temp);
        }
        return ans;
    }
};