快速排序算法

快速排序算法基础

快速排序算法原理

简单说，快速排序算法就是在给定数组中，找到各个元素所应该在的位置，即满足在这个元素之前的元素都应小于它，在它之后的都应大于它，如：

其中，找到该元素所在的正确位置和将元素移动到正确位置的功能都由partition方法实现

• 给partition方法传入三个参数arr(待排序数组),l(待排序数组左端),r(待排序数组右端)，返回元素正确位置的索引

整个过程的伪代码：

三路快速排序算法的实现

• 原理：将待排序数组分为三段，>v,<v和==v，用i来遍历整个数组，具体区间、循环不变量如下图
  
  实现三路快速排序：

package QuickSort;

import java.util.Random;

public class QuickSort2 {
	private QuickSort2() {
	}
	
	public static<E extends Comparable<E>> void sort3ways(E[] arr) {
		Random rnd=new Random();
		sort3ways(arr,0,arr.length-1,rnd);
	}
	
    private static<E extends Comparable<E>> void sort3ways(E[] arr,int l,int r,Random rnd) {
    	if(l>=r) return;
		int p=l+rnd.nextInt(r-l);
		swap(arr,l,p);
    	//[l+1,lt]<v;[lt+1,i-1]==v;[gt,r]>v
    	int i=l+1,lt=l,gt=r+1;
    	while(i<gt) {
    		if(arr[i].compareTo(arr[l])<0) {
    			lt++;
    			swap(arr,lt,i);
    			i++;
    		}
    		else if(arr[i].compareTo(arr[l])>0) {
    			gt--;
    			swap(arr,gt,i);
    		}
    		else
    			i++;
    	}
    	swap(arr,l,lt);
    	
    	sort3ways(arr,l,lt-1,rnd);
    	sort3ways(arr,gt,r,rnd);
    	
    	
	}
    
    private static <E extends Comparable<E>>  void swap(E[] arr,int i,int j) {
    	E tmp=arr[i];
    	arr[i]=arr[j];
    	arr[j]=tmp;
    }
    
    public static void main(String[] args) {
		Integer[] arr= {4,6,1,9,3,7,2,8,5};
		QuickSort2.sort3ways(arr);
		for(int i=0;i<arr.length;i++)
			System.out.print(arr[i]+"  ");
	}

}

结果：

SelectK问题

给出一个无序数组，找出第K小的元素
思路：由于快速排序算法的原理就是找到数字应该在的位置，我们其实就已经知道它在数组中是第几小的元素了

代码

package QuickSort;

import java.util.Random;

public class SelectK2 {
	private SelectK2() {
	}
	
	public static int select(int[] arr,int k) {
		Random rnd=new Random();
		return select(arr,0,arr.length-1,k-1,rnd);
	}
	
	private static int select(int[] arr,int l,int r,int k,Random rnd) {
		int p=partition(arr,l,r,rnd);
		if(p==k) return arr[p];
		
		if(p>k)return select(arr,l,p-1,k,rnd);
		else  return select(arr,p+1,r,k,rnd);
	}
	
	private static int partition(int[] arr,int l,int r,Random rnd) {
		int p=l+rnd.nextInt(r-l+1);
		swap(arr,p,l);
		
		int i=l+1,j=r;
		while(true) {
			while(i<=j&&arr[i]<arr[l]) {
				i++;
			}
			while(i<=j&&arr[j]>arr[l]) {
				j--;
			}
			if(i>=j)  break;
			
			swap(arr,i,j);
			i++;
			j--;
		}
		//i的索引为k+1,j为k
		swap(arr,l,j);
		return j;
	}
	
	private static void swap(int[] arr,int i,int j) {
		int tmp=arr[i];
		arr[i]=arr[j];
		arr[j]=tmp;
	}
	
	public static void main(String[] args) {
		int[] arr= {9,235,6,287,943};
		SelectK2 sk2=new SelectK2();
		System.out.println(sk2.select(arr, 3));
	}

}

结果：