C++如何实现类似Python的[]风格列表函数入参及参数遍历

实现说明 #

C++语法规定[]下标运算符必须作为类的成员函数重载，无法直接实现完全无前置符号的func(["a","b"])写法。我们可以通过不到20行的轻量代码，用一个单字符全局辅助对象实现和Python风格几乎一致的调用效果，无需使用大括号传参。

完整可运行代码 #

#include <vector>
#include <string>
#include <iostream>

// 辅助泛型包装类，用于拼接[]内的多参数
template<typename T>
struct ListBuilder {
    std::vector<T> data;
    // 重载逗号运算符，支持连续追加同类型参数
    ListBuilder& operator,(const T& val) {
        data.push_back(val);
        return *this;
    }
    // 隐式转换为vector，函数可直接接收vector类型参数
    operator std::vector<T>() const {
        return data;
    }
};

// 辅助空类，用于重载operator[]
struct ListBuilderDummy {
    template<typename T>
    ListBuilder<T> operator[](const T& first) const {
        ListBuilder<T> lb;
        lb.data.push_back(first);
        return lb;
    }
};

// 全局单字符辅助对象，使用时视觉上几乎无感知
constexpr ListBuilderDummy _;

// 你的业务函数，参数直接定义为对应类型的vector即可
void func(const std::vector<std::string>& args) {
    // 遍历逻辑见下文
    for (const auto& s : args) {
        std::cout << s << " ";
    }
    std::cout << std::endl;
}

int main() {
    // 调用示例，和Python风格几乎一致
    func(_["a","b","c"]);
    func(_["a","b","c","d"]);
    // 也支持其他同类型参数，比如int
    // func(_[1,2,3,4,5]);
    return 0;
}

参数遍历方法 #

下面提供三种常用的遍历方式，可按需选择：

• 范围for遍历（C++11及以上支持，最简洁，推荐使用）

for (const auto& item : args) {
    // 处理单个参数item
    std::cout << item << std::endl;
}

• 迭代器遍历

for (auto it = args.cbegin(); it != args.cend(); ++it) {
    // 处理单个参数*it
    std::cout << *it << std::endl;
}

• 下标遍历

for (size_t i = 0; i < args.size(); ++i) {
    // 处理单个参数args[i]
    std::cout << args[i] << std::endl;
}

扩展说明 #

• 支持任意同类型参数传递，比如int、自定义类等，仅需要修改函数入参的vector模板类型即可
• 最低支持C++11标准，当前所有主流编译器默认均支持该标准
• 无额外第三方依赖，仅需要引入STL的vector头文件

内容的提问来源于stack exchange，提问作者Alejandro