使用 STL std::sort 作为 qsort_r

Question

通过qsort_r，我们可以在比较函数中使用外部参数。例如，

int cmp_with_parameter(void *a, void *b, void *p) 
{
  return (int*)p[*(int*)a] - (int*)p[*(int*b)];
}

int main() {
  int values[] = {1,2,3,4};
  int arr[] = {0, 1, 2, 3};
  qsort_r(arr, 4, sizeof(arr[0]), cmp_with_parameter, values);
}

我们可以按照values[]中的值对int arr[]进行排序。

现在，我想用 std::sort 实现类似的功能，这就是我如何在 std::sort 中使用外部值？先感谢您。

Answer 1

std::sort()所采用的比较函数可以是一个函数对象，它保存着合适的信息。例如：

std::sort(std::begin(arr), std::end(arr),
          [=](int a, int b){ return values[a] < values[b]; });

Answer 2

你需要创建一个 functor 作为你的比较器。制作 functor 的常用方法是使用 struct 并重载调用运算符：

struct cmp_with_parameter
{
    const int* values; // keep a pointer to the external array

    cmp_with_parameter(const int* values): values(values) {}

    // overloading this operator allows an object
    // of this struct to be called like a function.
    bool operator()(int a, int b) const
    {
        return values[a] < values[b];
    }
};

int main()
{
    int values[] = { 1, 2, 3, 4 };
    int arr[] = { 0, 1, 2, 3 };
    std::sort(arr, arr + sizeof(arr)/sizeof(arr[0]), cmp_with_parameter(values));

    // ...
}

如果你的编译器支持C++11，你可以简化std::sort()参数：

int main()
{
    int values[] = { 1, 2, 3, 4 };
    int arr[] = { 0, 1, 2, 3 };
    std::sort(std::begin(arr), std::end(arr), cmp_with_parameter(values));

    // ...
}

维基百科有一个很好的例子来说明如何解决这个问题：Function Object:C/C++