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

最新推荐文章于 2023-06-07 23:54:19 发布

肩上杯雪

最新推荐文章于 2023-06-07 23:54:19 发布

阅读量191

点赞数 1

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/qq_39763472/article/details/109742186

版权

LeetCode 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

该博客详细介绍了LeetCode第1030题的解决方案，通过两种方法解决如何按曼哈顿距离升序排列矩阵单元格的问题。方法一是遍历并计算所有点的距离，利用C++11的lambda表达式进行排序；方法二是利用几何特性，枚举曼哈顿距离并在斜正方形上顺时针遍历。

摘要由CSDN通过智能技术生成

题意
给定一个矩形，row ∈ [0, R) col ∈ [0, C)，给定一个点 (r0, c0)，返回一个vector<vector< int > >，其中每一个 vector< int > 是一对坐标点，顺序按照这些点到 (r0, c0) 距离的曼哈顿距离进行升序排列。
思路

方法一
遍历矩形中的所有点，计算每个点到给定点的曼哈顿距离，sort即可。

class Solution {
public:
    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        vector<vector<int>> ret;
        for (int i = 0; i < R; i++) {
            for (int j = 0; j < C; j++) {
                ret.push_back({i, j});
            }
        }
        sort(ret.begin(), ret.end(), [=](vector<int>& a, vector<int>& b) {
            return abs(a[0] - r0) + abs(a[1] - c0) < abs(b[0] - r0) + abs(b[1] - c0);
        });
        return ret;
    }
};

其中 sort内部的 [=] 是 c++ 11引入的lambda 表达式。
lambda 表达式其实就是一个语法糖，具体可以参看以下链接：
https://www.jianshu.com/p/d686ad9de817
在这里插入图片描述

方法二
几何方法，上述方法枚举的是矩阵中的所有点，此方法是枚举曼哈顿距离。可以画图发现，曼哈顿距离相同的点都在以 (r0, c0) 为中心的斜正方形上。

从图中可以看出，曼哈顿距离从1开始枚举，每次枚举时，正方形从左顶点开始沿顺时针遍历完此正方形上的所有点。

class Solution {
public:
    // 几何法，枚举曼哈顿距离
    vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
        int dir[4][2] = {
            {1, 1},
            {1, -1},
            {-1, -1},
            {-1, 1}
        };
        vector<vector<int>> ans;
        // 最大的曼哈顿距离
        int maxDist = max(r0, R - r0 - 1) + max(c0, C - c0 - 1);
        int row = r0, col = c0;
        ans.push_back({r0, c0});
        for(int dist = 1; dist <= maxDist; dist ++){
            row --;
            for(int i = 0;i < 4;i ++){
                // 遍历斜正方形的边上所有点
                while((i % 2 == 0 && r0 != row) || (i % 2 != 0 && c0 != col)){
                    // 矩形边界条件
                    if(row >= 0 && row < R && col >= 0 && col < C){
                        ans.push_back({row, col});
                    }
                    row += dir[i][0];
                    col += dir[i][1];
                }
            }
        }
        return ans;
    }
};