package arrays;

/**
 * 利用快速排序排序数组
 */
public class array_25 {
    /**
     * o(nlog(n)),最坏情况为 o(n^2)
     */
    public int[] solution(int[] arr){
        return quickSort(arr, 0, arr.length - 1);
    }

    private int[] quickSort(int[] arr, int left, int right) {
        if (left < right) {
            int partitionIndex = partition(arr, left, right);
            quickSort(arr, left, partitionIndex - 1);
            quickSort(arr, partitionIndex + 1, right);
        }
        return arr;
    }

    private int partition(int[] arr, int left, int right) {
        // 设定基准值（pivot）
        int pivot = left;
        int index = pivot + 1;
        //设立基准值，把小于基准值的放在左边，大于的放到右边，小于index+1位置的数都小于等于pivot
        for (int i = index; i <= right; i++) {
            if (arr[i] < arr[pivot]) {
                swap(arr, i, index);
                //说明出现了一个比pivot小的数了，放到左边，然后index往后移动一个准备放下一个小于的。
                index++;
            }
        }
        swap(arr, pivot, index - 1);
        return index - 1;
    }

    private void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }

}