LeetCode第 1030 题：距离顺序排列矩阵单元格(C++)

最新推荐文章于 2021-12-18 06:40:51 发布

zj134_

最新推荐文章于 2021-12-18 06:40:51 发布

阅读量261

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_32523711/article/details/109737866

leetcode 专栏收录该内容

334 篇文章

订阅专栏

博客围绕力扣题目“距离顺序排列矩阵单元格”展开，指出这本质是定义排序规则的问题。介绍了使用优先级队列和排序的方法，但复杂度较高，还提及因数据量不大，桶排序分桶较容易。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1030. 距离顺序排列矩阵单元格 - 力扣（LeetCode）
在这里插入图片描述
其实就是定义一个排序规则的问题

使用优先级队列

class Solution {
public:
    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        vector<vector<int>> res;
        //https://blog.csdn.net/qq_32523711/article/details/108226095
        auto cmp = [&r0, &c0](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);};
        priority_queue<vector<int>, vector<vector<int>>, decltype(cmp)> q(cmp);
        for(int i = 0; i < R; ++i){
            for(int j = 0; j < C; ++j)  q.push({i, j});
        }
        while(!q.empty()){
            res.push_back(q.top());
            q.pop();
        }
        return res;
    }
};

排序

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(), [&r0, &c0](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:
    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        int maxDis = max(r0, R-1-r0) + max(c0, C-1-c0);
        vector<vector<int>> bucket[maxDis+1];
        for(int i = 0; i < R; ++i){
            for(int j = 0; j < C; ++j){
                int dis = abs(i-r0) + abs(j-c0);
                bucket[dis].push_back({i, j});
            }
        }
        vector<vector<int>> res;
        for(int i = 0; i <= maxDis; ++i){
            for(auto &v : bucket[i])  res.push_back(v);
        }
        return std::move(res);
    }
};