题目:
今天这道题和我第一天刷的题很像,基本都是可以用自定义排序解决的。
于是就有了最简单的方法(使用sort自定义排序)
C++代码如下(带测试)
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
vector<vector<int>> Point;
for(int i = 0;i<R;i++)
{
for(int j=0;j<C;j++)
{
Point.push_back({i,j});
}
}
sort(Point.begin(),Point.end(),[=](vector<int>& u, vector<int>& v) {//Lambda表达式
return abs(u[0] - r0) + abs(u[1] - c0) < abs(v[0] - r0) + abs(v[1] - c0);
});
return Point;
}
};
int main()
{
vector<vector<int > > Point;
Solution s;
Point = s.allCellsDistOrder(2,3,1,2);
for(int i=0;i<6;i++)
{
cout<<Point[i][0]<<","<<Point[i][1]<<endl;
}
}
多提一句,中间Sort方法第三个参数的写法使用的是Lambda表达式
具体的可以参考下面这篇博文
C++ lambda表达式入门
后面的那个几何法,很好理解,也很巧妙,时间效率貌似还不错,我就直接把代码贴过来了
class Solution {
public:
const int dr[4] = {1, 1, -1, -1};
const int dc[4] = {1, -1, -1, 1};
vector<vector<int>> allCellsDistOrder(int R, int C, int r0, int c0) {
int maxDist = max(r0, R - 1 - r0) + max(c0, C - 1 - c0);
vector<vector<int>> ret;
int row = r0, col = c0;
ret.push_back({row, col});
for (int dist = 1; dist <= maxDist; dist++) {
row--;
for (int i = 0; i < 4; i++) {
while ((i % 2 == 0 && row != r0) || (i % 2 != 0 && col != c0)) {
if (row >= 0 && row < R && col >= 0 && col < C) {
ret.push_back({row, col});
}
row += dr[i];
col += dc[i];
}
}
}
return ret;
}
};
//作者:LeetCode-Solution
//链接:https://leetcode-cn.com/problems/matrix-cells-in-distance-order/solution/ju-chi-shun-xu-pai-lie-ju-zhen-dan-yuan-ge-by-leet/
//来源:力扣(LeetCode)
//著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。