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

时间:2015-06-16 13:41:22

标签: c++ sorting stl compare

在下面的代码中,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;

}

1 个答案:

答案 0 :(得分:3)

将会员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
    );