在刷Leetcode时,我遇到了一道题目(详见Leetcode 第426场周赛分析总结Q3),需要对两棵树建图,然后以每个节点作为根节点进行DFS遍历。一般的实现方法是将重复的逻辑封装起来,写两个函数,一个负责建图,另一个负责DFS,然后将建图后的返回值作为参数传递给DFS。
在Python、JavaScript等高级语言中,有一种叫做闭包函数的编程技巧,能够简化这种逻辑。闭包本质上是一个函数A返回另一个函数B,而函数B捕获了函数A的局部变量,从而使函数B拥有状态信息。在C++中,我们也可以利用闭包函数,将建图与DFS的逻辑绑定,从而避免显式地在调用时传递参数。
以下是该问题的C++实现:
class Solution {
public:
vector<int> maxTargetNodes(vector<vector<int>>& edges1, vector<vector<int>>& edges2, int k) {
// 建立一个闭包函数,用于返回DFS函数
auto get_dfs = [](decltype(edges1) edges) -> auto {
int n = edges.size() + 1;
vector graph(n, vector<int>());
for (auto &&edge : edges) {
auto u = edge