c++ ——排序并保留索引

shuzfan

于 2016-09-30 17:53:05 发布

阅读量1.9w

点赞数 18

CC 4.0 BY-SA版权

分类专栏： C++

本文链接：https://blog.youkuaiyun.com/shuzfan/article/details/52712923

C++ 专栏收录该内容

16 篇文章

订阅专栏

本文介绍如何在C++中实现类似于Matlab的排序功能，不仅返回排序后的元素，还返回原始元素的位置索引。通过使用C++11的lambda表达式和STL中的sort与iota函数，实现了一种高效的方法来获取排序后的索引。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

C++自带的排序函数sort只能获得排好序的值，没办法像Matlab一样同时返回排序索引。

下面使用c++11中的 lambda来为sort函数添加一个模式：

template <typename T>
vector<size_t> sort_indexes(const vector<T> &v) {

  // 初始化索引向量
  vector<size_t> idx(v.size());
  //使用iota对向量赋0~？的连续值
  iota(idx.begin(), idx.end(), 0);

  // 通过比较v的值对索引idx进行排序
  sort(idx.begin(), idx.end(),
       [&v](size_t i1, size_t i2) {return v[i1] < v[i2];});
  return idx;
}

上面的是升序排列，如果想要降序排列，只需要更改为：

[&v](size_t i1, size_t i2) {return v[i1] > v[i2];});