C++11 实现 argsort

看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的闭包特性(访问非全局的外部变量),单独定义一个函数的话可能涉及到全局变量,很麻烦。