c++ finding the argsort of a vector

Question

Following C++ sorting and keeping track of indexes : This is how to get the argsort of vector<int> v

vector<int> VectorArgSort(vector<int> &v) {
    /* Get the indices that gives a sorted vector v*/

    vector<int> retIndices(v.size());
    iota(retIndices.begin(), retIndices.end(), 0);

    stable_sort(
        retIndices.begin(), retIndices.end(),
        [&v](int i1, int i2) {return v[i1] < v[i2];});

    return retIndices;
}

But what i want is to have vector<int> VectorArgSort(vector<float> &v) that still gives vector because they are indices, but to perform it on vector<float> &v instead.

I tried changing several of the defined int types to doubles, but i dont think i should because they all points towards the indices. So im not sure how to change the function to get what i want.

Please show the error message then. – Lukas-T Dec 06 '20 at 14:27 — Lukas-T, Dec 06 '20 at 14:27

score 0 · Answer 1 · answered Dec 06 '20 at 19:28

0

You only have to change the vector<int> to vector<float> in the argument of the function.

vector<int> VectorArgSort(vector<float> &v) {

see working version here

answered Dec 06 '20 at 19:28

Thomas

4,980
2
15
30

c++ finding the argsort of a vector

1 Answers1