题目描述 [ 距离顺序排列矩阵单元格]

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

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

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

示例

输入：R = 2, C = 2, r0 = 0, c0 = 1
输出：[[0,1],[0,0],[1,1],[1,0]]
解释：从 (r0, c0) 到其他单元格的距离为：[0,1,1,2]
[[0,1],[1,1],[0,0],[1,0]] 也会被视作正确答案。

解题思路一

• 广度优先搜索
• 要把xy坐标都存下来，所以略微复杂了点把

代码

class Solution {
public:
    void push(int i, int j, queue<pair<int, int> > &queue1, int R, int C){
        if(i<0||i>R-1||j<0||j>C-1) return;
        else{
            queue1.push(make_pair(i, j));
        }
    }

    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        vector<vector<int> > res;
        queue<pair<int, int> > queue1;
        set<pair<int, int> > hashset;
        if(R==0 || C==0) return res;
        queue1.push(make_pair(r0, c0));
//        hashset.insert(make_pair(r0, c0));
        while(!queue1.empty()){
            auto temp = queue1.front();
            queue1.pop();
            if(hashset.count(temp)==0){
                hashset.insert(temp);
                vector<int> ans;
                ans.push_back(temp.first);
                ans.push_back(temp.second);
                res.push_back(ans);
                push(temp.first-1, temp.second, queue1, R, C);
                push(temp.first+1, temp.second, queue1, R, C);
                push(temp.first, temp.second-1, queue1, R, C);
                push(temp.first, temp.second+1, queue1, R, C);
            }
        }
        return res;
    }
};

解题思路二

• 把矩阵中的点到目标点的距离按照距离进行排序输出即可
• 大概是脑残所以才会想到上面那么复杂的解法把

代码

class Solution {
public:
    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        vector<vector<int>> res(R*C,vector<int>(3));
        int num=0;
        for(int i=0;i<C;i++)
        {
            for(int j=0;j<R;j++)
            {
                res[num][0]=j;
                res[num][1]=i;
                res[num][2]=abs(r0-j)+abs(c0-i);
                num++;
            }
        }
        sort(res.begin(),res.end(),ismax);//排序
        for(int i=0;i<num;i++)//将曼哈顿距离删除
        {
            res[i].pop_back();
        }
        return res;
    }
    static bool ismax(vector<int> &a,vector<int> &b)
    {
        return a[2]<b[2];
    }
};