c++ lambda的匿名特性,无法直接在lambda内部递归调用自身。
借助std::function,这是一个可调用对象包装器,可以容纳除了类成员函数指针以外所有的可调用对象,适合作为回调函数使用。
Leetcode 47 全排列II
题目:
给定一个可包含重复数字的序列 nums ,按任意顺序 返回所有不重复的全排列。
第一种写法,直接用function包装,lambda通过[&]捕获引用,进而实现递归调用,缺点就是这种写法一定要有个返回值类型,对于void要改成返回bool。
class Solution {
public:
vector<vector<int>> permuteUnique(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<vector<int>> ans;
vector<int> temp(nums.size());
bitset<8> b;
std::function<bool()> DFS = [&]()->bool {
int index = b.count();
for (size_t i = 0; i < nums.size(); i++) {
if (!b[i]) {
if (i && !b[i - 1] && nums[i] == nums[i - 1]) {
continue;
}
temp[index] = nums[i];
if (index + 1 == nums.size())
ans.push_back(temp);
else {
b[i] = true;
DFS();
}
b[i] = false;
}
}
return false;
};
DFS();
ans.shrink_to_fit();
return ans;
}
};
第二种,把lambda函数在外部用function包装,在调用lambda时把新函数作为参数传入(给冰箱起个别名“大象”,然后把“大象”装进冰箱里)
class Solution {
public:
vector<vector<int>> permuteUnique(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<vector<int>> ans;
vector<int> temp(nums.size());
bitset<8> b;
auto dfs = [&](auto&& me)->void {
int index = b.count();
for (size_t i = 0; i < nums.size(); i++) {
if (!b[i]) {
if (i && !b[i - 1] && nums[i] == nums[i - 1]) {
continue;
}
temp[index] = nums[i];
if (index + 1 == nums.size())
ans.push_back(temp);
else {
b[i] = true;
me(me);
}
b[i] = false;
}
}
};
// 包装
std::function<void(decltype(dfs))> DFS = dfs;
// 作为参数传入
DFS(dfs);
ans.shrink_to_fit();
return ans;
}
};