std::sort 使用索引但保持索引顺序

Question

我有一个值向量。我想根据这些值得到一个排序的索引列表。

除了出现相同值的情况外，我已经很好地完成了这项工作。当出现相同的值时，我希望索引保持有序。

例如我有这个测试用例：

std::vector< size_t >   idx;
std::vector< int >      val;
for( int i = 0; i < 40; i++ )
{
    idx.push_back( i );
    val.push_back( i % 10 );
}

std::sort( idx.begin(), idx.end(), [&]( size_t a, size_t b )
    {
        return val[a] < val[b];
    } );

这会将索引数组排序为以下内容：

(0,10,30,20,1,31,21,11,2,22,12,32,3,13,23,33,4,14,24,34,5,15,25,35,6,16,26,36,7,17,27,37,8,28,18,38,9,29,19,39)

但我希望数组按以下顺序排列：

(0,10,20,30,1,11,21,31,2,12,22,32,3,13,23,33,4,14,24,34,5,15,25,35,6,16,26,36,7,17,27,37,8,18,28,38,9,19,29,39)

有没有一种简单的方法可以修改我的 lambda 以按照最后一个指定的顺序获取这些值？

提前干杯！

Answer 1

您应该使用std::stable_sort。

在您的特定情况下，您可以调整 lambda 以将 a 与 b 与 val[a] == val[b] 进行比较，但这会使您的意图对任何可能偶然发现此代码的未来开发人员变得模糊。

Answer 2

当 val[a] == val[b] 时，返回两个索引中较小的一个。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> val;
int compare(int a,int b)
{
    if(val[a] == val[b]) {
        return a < b;
    }
    return val[a] < val[b];
}
int main()
{
    vector<int> idx;
    int i;
    for(i = 0; i < 40; i++) {
        idx.push_back(i);
        val.push_back(i%10);
    }
    sort(idx.begin(),idx.end(),compare);
    for(i = 0; i < 40; i++) {
        cout<<idx[i]<<' ';
    }
}