将R split()函数转换为C++

#include <iostream>
#include <vector>
#include <map>

// 定义一个函数，将向量按照指定的因子进行分组
std::map<std::string, std::vector<int>> split(const std::vector<int>& vec, const std::vector<std::string>& factor) {
    std::map<std::string, std::vector<int>> result;

    // 遍历向量
    for (int i = 0; i < vec.size(); i++) {
        // 获取当前元素的因子
        std::string current_factor = factor[i];

        // 将当前元素添加到对应因子的向量中
        result[current_factor].push_back(vec[i]);
    }

    return result;
}

int main() {
    // 示例数据
    std::vector<int> vec = {1, 2, 3, 4, 5};
    std::vector<std::string> factor = {"A", "B", "A", "B", "A"};

    // 调用split函数进行分组
    std::map<std::string, std::vector<int>> result = split(vec, factor);

    // 打印结果
    for (const auto& pair : result) {
        std::cout << "Factor: " << pair.first << std::endl;
        std::cout << "Values: ";
        for (const auto& value : pair.second) {
            std::cout << value << " ";
        }
        std::cout << std::endl;
    }

    return 0;
}