如何在 C++ STL 的 sort() 中编写 Comp 函数

Question

在下面的代码中，strArray是一个字符串数组，vec包含了每个字符串对应的ID。 我想根据 strArray 中的字符串对 vec 中的 ID 进行排序。

如何设计排序中的 Comp 函数？

class myClass
{
public:
        int V;
        vector<int> vec;
        string* strArray;

        myClass(int v);
        void myFunc();
}

myClass::myClass(int v, vector<int> vec1)
{
    V = v;
    strArray = new string[v];
    vec.swap(vec1)
}

void myClass::myFunc()
{
    //....

    for(i = 0; i<V; i++)
        strArray[i] = GenerateString(vec[i]);// GenerateString() can return a string.

    sort(vec.begin(), vec.end(), Comp);// sort vec based on strArray,how to design the Comp funtion?

    //....
}

main()
{
    int myints[] = {32,71,12,45,26,80,53,33};
    vector<int> myvector (myints, myints+8);

    myClass obj(8, myvector);

    obj.myFunc();

    return 0;

}

Answer 1

将成员string* strArray; 更改为std::pair<std::string, int> strArray;，更改分配strArray = new std::pair<std::string, int>[v]; 并在您的myFunc 中进行以下更改

void myClass::myFunc()
{
    //....

    for(i = 0; i<V; i++)
        strArray[i] = make_pair(GenerateString(vec[i]), vec[i]);// GenerateString() can return a string.

    sort(vec.begin(), vec.end());
    std::transform(
        strArray,
        strArray + V,
        vec.begin(),
        [](const std::pair<std::string, int> &p) -> int { return p.second; }
    );

    //....
}

在这段代码中，我们将生成字符串的数字 (vec[i]) 配对或配对，并对数组进行排序。现在使用std::transform，将号码复制回vec

来自评论

如果你不懂 lambda 函数，你也可以在别处写一个函数，作为第四个参数传入std::transform

static int fetch_int_from_pair(const std::pair<std::string, int> &p) {
 return p.second;
}
...
td::transform(
        strArray,
        strArray + V,
        vec.begin(),
        fetch_int_from_pair
    );