c++ - 绝对值排序，使用自定义比较器

Question

背景:

我在一次 pramp 面试中被问到这个问题，我无法通过所有测试用例。

问题:

Absolute Value Sort

Given an array of integers arr, write a function absSort(arr), that sorts the array according to the absolute values of the numbers in arr. If two numbers have the same absolute value, sort them according to sign, where the negative numbers come before the positive numbers.

Examples:

input: arr = [2, -7, -2, -2, 0]

output: [0, -2, -2, 2, -7]

我的尝试:

  std::sort(nums.begin(), nums.end(), [](int a, int b) {
    if(abs(a) < abs(b))
      return true;
    return a < b ? (a < 0 && abs(a) > b) : (b < 0 && abs(b) > a);
  });

例如，我通过了一些测试用例，但不是全部

Input: [2,-7,-2,-2,0]
Expected: [0,-2,-2,2,-7]
Actual: [0, -2, -7, 2, -2 ]

我觉得我只需要对我的 lambda 函数做一些小的调整，但我想不出来。

Answer 1

这可能有用

    std::sort(nums.begin(), nums.end(), [](int a, int b) {
    if(abs(a) != abs(b))
      return abs(a) < abs(b);
    return a < b;
  });