c++查找vector<float>的argsort

#include <iostream>
#include <vector>
#include <algorithm>

template <typename T>
std::vector<size_t> argsort(const std::vector<T>& v) {
    // 初始化索引向量
    std::vector<size_t> idx(v.size());
    for (size_t i = 0; i < idx.size(); ++i) {
        idx[i] = i;
    }

    // 使用lambda表达式对索引进行排序
    std::sort(idx.begin(), idx.end(),
              [&v](size_t i1, size_t i2) { return v[i1] < v[i2]; });

    return idx;
}

int main() {
    std::vector<float> vec = {3.14, 2.71, 1.41, 1.73, 2.23};

    // 调用argsort函数获取排序后的索引
    std::vector<size_t> sortedIdx = argsort(vec);

    // 输出排序后的索引
    for (size_t i = 0; i < sortedIdx.size(); ++i) {
        std::cout << sortedIdx[i] << " ";
    }
    std::cout << std::endl;

    return 0;
}