C++ lamda表达式

leetcode1030 距离顺序排序矩阵单元格

Q：

给定四个整数 rows ,   cols ,  rCenter 和 cCenter 。有一个 rows x cols 的矩阵，你在单元格上的坐标是 (rCenter, cCenter) 。

返回矩阵中的所有单元格的坐标，并按与 (rCenter, cCenter) 的 距离 从最小到最大的顺序排。你可以按 任何 满足此条件的顺序返回答案。

单元格(r1, c1) 和 (r2, c2) 之间的距离为|r1 - r2| + |c1 - c2|。

A：

看到题目第一反应是用bfs，一层一层搜索上下左右四个节点，但显然计算量很大，要判断节点是否越界和是否已被访问。

如果用sort()函数排序，则会快很多，用lamda表达式可以简化代码

class Solution {
public:
    vector<vector<int>> allCellsDistOrder(int rows, int cols, int rCenter, int cCenter) {
        vector<vector<int>> res;

        for(int i=0;i<rows;i++){
            for(int j=0;j<cols;j++){
                res.push_back({i,j});
            }
        }

        auto dist = [&](int x, int y){
            return abs(x-rCenter) + abs(y-cCenter);
        };

        sort(res.begin(),res.end(),
        [&](const vector<int>& a, const vector<int>& b){
            return dist(a[0],a[1]) < dist(b[0],b[1]);
        }
        );
        return res;
    }
};

lamda表达式的基本形式为：

[capture list] (params list) mutable exception-> return type { function body }

其中的一些元素可以被省略，最常用的形式为：

[capture list] (params list) {function body}

[capture list]表示捕获外部变量的方式，也就是外部变量如何传进function body

[variance_name]：值捕获

[&variance_name]：引用捕获

[=]: 隐式捕获，方式为值捕获，不需要写明外部变量名字

[&]: 隐式捕获，方式为引用捕获