C++ lambda递归

2 篇文章 0 订阅

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;
	}
};
  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值