partition是快速排序的核心。
目的:将数组划分为<=pivot和>pivot两部分,或者是<pivot和>=pivot。
实现:单向扫描,或者双向扫描。
方法一:单向扫描
思路:两个指针,i 用来遍历数组,与pivot进行比较;small用来记录小于pivot的位置+1。for (i),当 i 的值小于pivot时,将其与small的值进行交换,small++。
需要注意的是,small本身的值>povit。
/**
* <p>将数列以某值进行划分,前面小于该值,后面大于<p>
* @param pivotIndex: 划分值
* */
public int partition(int[] nums, int start, int end, int pivotIndex) {
int pivot = nums[pivotIndex];
nums[pivotIndex] = nums[end];
nums[end] = pivot;
int small = start;
// small用来记录大于pivot的位置
System.out.println(Arrays.toString(nums));
for (int i=start; i<end; i++) {
// 如果i>pivot,small的位置不动,直到找到i小于pivot
if (nums[i] < pivot) {
// small<i,small是大于pivot,i是小于pivot
int temp = nums[i];
nums[i] = nums[small];
// 如果small==i,等同不变
nums[small] = temp;
++small;
}
System.out.println(i + "--"+small+"--"+pivot);
System.out.println(Arrays.toString(nums));
System.out.println("");
}
int s = nums[small];
nums[small] = nums[end];
nums[end] = s;
System.out.println(Arrays.toString(nums));
return small;
}
方法二:双向扫描