计数排序/排序算法的变体

Question

有一个问题，它给了我一个随机数作为枢轴，我必须将我的数组W.R.T排序到这个枢轴(最接近的先，然后最远)例如

array =[2,7,4,6,4,4,5,3,6,9,1,1,9] and 
    pivot=5 
    expected output: [5,4,4,6,6,3,7,2,1,1,9,9]

这是计数排序的一种变体吗?如果不是!谁能给我一个解决这个问题的提示吗?在思考如何处理计数和数组索引时，我遇到了一个障碍因此，到目前为止，我已经能够做到这一点

class HelloEclipse{
public static void main(String[] args) {
    Scanner sc=new Scanner(System.in);
    int N=sc.nextInt();
    int pivot=sc.nextInt();
    int[] mainArray=new int[N];
    int[] differenceArray=new int[N];
    int[] differnceCountArray=new int[Integer.MAX_VALUE];
    for(int i=0;i<N;i++){
        mainArray[i]=sc.nextInt();
        differenceArray[i]=pivot-mainArray[i];
        if(differenceArray[i]>0){
        differnceCountArray[differenceArray[i]]++;}
        else{
            differnceCountArray[-differenceArray[i]]++;
        }
    }
    }
}

任何关于如何进行的建议将是有帮助的!

Answer 1

编写一个合适的整数比较器，并使用arrays .sort:

public class PivotComparator implements Comparator<Integer> {
    private int pivot;
    public PivotComparator(int pivot) {
        super();
        this.pivot = pivot;
    }
    @Override
    public int compare(Integer a, Integer b) {
        return Math.abs(a - pivot) - Math.abs(b - pivot);
    }
    public static void main(String[] args) {
        Integer[] toSort = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
        Comparator<Integer> comp = new PivotComparator(5);
        Arrays.sort(toSort, comp);
        for (Integer i : toSort) {
            System.out.println(i);
        }
    }
}

编辑

要把所有的4放在6的前面，你可以这样做(而不是排序两次)

public int compare(Integer a, Integer b) {
    int diff = Math.abs(a - pivot) - Math.abs(b - pivot);
    if (diff != 0) {
        return diff;
    }
    return a - b;
}

Answer 2

这是一个实现

    int[] array = { 2, 7, 4, 6, 4, 4, 5, 3, 6, 9, 1, 1, 9 };
    int pivot = 5;
    int[] sorted = Arrays.stream(array).boxed().sorted().sorted((i1, i2) -> Math.abs(i1 - pivot) - Math.abs(i2 - pivot)).mapToInt(i -> i).toArray();
    System.out.println(Arrays.toString(sorted));

输出

[5, 4, 4, 4, 6, 6, 3, 7, 2, 1, 1, 9, 9]