Kotlin练习-快速排序

Java版本

object QuickSort {

    fun sort(array: MutableList<Int>): MutableList<Int> {

        if (array.size == 0 || array.size == 1) return array
        val flag = array[0]
        //删除基准元素，如果不删除的话，基准值也会进入左边篮子中，而且永远是第一个位置，第二次开始，所有元素都小于4，就会进入死循环。
        array.removeAt(0)
        val leftArray = mutableListOf<Int>()
        val rightArray = mutableListOf<Int>()
        //站队，小于等于基准值的放到左边篮子里，大于基准值放到右边篮子里
        array.forEach {
            if (it <= flag) {
                leftArray.add(it)
            } else {
                rightArray.add(it)
            }
        }
        //递归排序左右两侧数据
        val sortedLeftArray = sort(leftArray)
        val sortedRightArray = sort(rightArray)

        //排序完成后,组装
        val sortedArray = mutableListOf<Int>()
        sortedLeftArray.forEach {
            sortedArray.add(it)
        }
        sortedArray.add(flag)
        sortedRightArray.forEach {
            sortedArray.add(it)
        }

        return sortedArray
    }

}