Pytorch学习---基于经典网络架构ResNet训练花卉图像分类模型-CFANZ编程社区

从前序与中序遍历序列构造二叉树

官方解析实在是记不住，翻别人的题解发现了一个有意思的写法

class Solution {
public:
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
        auto dfs = [](auto&& dfs, auto&& lP, auto&& lI, auto&& rI) {
            if (lI == rI) return (TreeNode*)nullptr;
            auto loc = find(lI, rI, *lP);
            return new TreeNode(*lP, dfs(dfs, lP + 1, lI, loc),
                                     dfs(dfs, lP + (loc - lI) + 1, loc + 1, rI));
        };
        return dfs(dfs, preorder.cbegin(), inorder.cbegin(), inorder.cend());
    }
};

看起来简单有意思，但是这个 auto dfs = [](auto&& dfs, 是什么意思呢？

lambda 自调用

C++11，借助std::function

#include <iostream>
#include <functional>
int main(int argc, char* argv[])
{
	std::function<int(int)> fib = [&fib](int n) { return n < 2 ? n : fib(n - 1) + fib(n - 2); };
	std::cout << fib(5);
	return 0;
}

借助 std::function 代码修改如下：

class Solution {
    public:
        TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
            function<TreeNode* (vector<int>::const_iterator, vector<int>::const_iterator, vector<int>::const_iterator)> 
                dfs = [&dfs]( auto&& lP, auto&& lI, auto&& rI) {
                if (lI == rI) return (TreeNode*)nullptr;
                auto loc = find(lI, rI, *lP);
                return new TreeNode(*lP, dfs( lP + 1, lI, loc),
                    dfs(lP + (loc - lI) + 1, loc + 1, rI));
            };
            return dfs( preorder.cbegin(), inorder.cbegin(), inorder.cend());
        }
    };

这个function 是个啥？没用过

C++11 std::function 基础用法

std::function是C++11标准库中提供的一种可调用对象的通用类型，它可以存储任意可调用对象，如函数指针，函数对象，成员函数指针和lambda表达式。std::function类模板是一个类似于函数指针的类型，但它是可以处理任意可调用对象的，并且可以检查调用对象是否为空。

需要头文件： functional

基本用法：

std::function对象可以像普通函数一样调用，并且可以使用bool类型的运算符来检查调用对象是否为空。

std::function<int(int, int)> f;
if (f)
    std::cout << f(1, 2) << std::endl;
else
    std::cout << "f is empty" << std::endl;

std::function可以存储智能指针，避免内存泄漏：

这段代码定义了一个变量add，它是一个std::function类型，这种类型可以存储一个可调用的函数（可以是函数指针、函数对象、lambda表达式等）。该函数的签名为int(int, int)，即返回值类型为int,接受两个int类型参数。变量add被赋值为一个指向匿名函数的指针。这个匿名函数接受两个int类型参数，并返回它们的和。使用std::make_shared<int(*)(int, int)>来创建该函数的共享指针

存储成员函数指针

调用类的成员函数：

class A {
public:
    int add(int a, int b) { return a + b; }
};
std::function<int(A&, int, int)> add = &A::add;
A a;
std::cout << add(a, 3, 4) << std::endl;

补充：iterator与const_iterator及const iterator区别

如果你传递过来一个const类型的容器，那么只能用const_iterator来遍历。

void Method(const vector<int> vInt)
{
  vector<int>::const_iterator iter;
}

中序与后序遍历序列构造二叉树

106. 从中序与后序遍历序列构造二叉树 - 力扣（LeetCode）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* buildTree(vector<int>& inorder, vector<int>& postorder) {
        function<TreeNode* (vector<int>::const_reverse_iterator, vector<int>::const_iterator, vector<int>::const_iterator)> 
                dfs = [&dfs]( auto&& lP, auto&& lI, auto&& rI) {
                if (lI == rI) return (TreeNode*)nullptr;
                auto loc = find(lI, rI, *lP);
                return new TreeNode(*lP, dfs( lP + (rI - loc), lI, loc),
                    dfs(lP + 1, loc + 1, rI));
            };
            return dfs( postorder.crbegin(), inorder.cbegin(), inorder.cend());
    }
};

注意两点： 1，vector<int>::const_reverse_iterator, postorder.crbegin() ，逆序迭代器

2，new TreeNode(*lP, dfs( lP + (rI - loc), lI, loc),
dfs(lP + 1, loc + 1, rI)); 计算移动位置与前序不同

参考：C++ 实现lambda递归调用（C++11 - C++23）_c++ lamda 递归-CSDN博客

C++11 std::function 基础用法_std::function用法-CSDN博客

iterator与const_iterator及const iterator区别