看python发现有这么个api,感觉很实用,想着stl里会不会有这个呢?查了半天毫无结果。于是用lambda自己实现了下。
以vector为例
template<typename T>
vector<int> argsort(const vector<T>& a){
int Len = a.size();
vector<int> idx(Len, 0);
for(int i = 0; i < Len; i++){
idx[i] = i;
}
std::sort(idx.begin(), idx.end(), [&v](int i1, int i2){return v[i1] < v[i2];});
return idx;
}
这个关键用到的是lambda的闭包特性(访问非全局的外部变量),单独定义一个函数的话可能涉及到全局变量,很麻烦。
时间: 2024-11-20 17:28:16