알고리즘 서론 학습 노트 - 배열 의 i 작은 요 소 를 찾 습 니 다.

/**
 *   ：    n                i         
 *   ：                       ，     ，              ，
 *                  。
 *         ：O(n^2),           O(n)
 */
public class RandomizedSort {

	/**
	 *  arr      ，                     ，    ，                 ，       
	 * @param arr    
	 * @param p       
	 * @param r       
	 * @param i    i      
	 */
	public int randomizedSort(int arr[],int p,int r,int i){
		if(p==r)
			return arr[p];
		// arr      ，arr[q]    arr  q-p+1      
		int q = partition(arr,p,r);
		int k = q-p+1;
		//  i  k，  arr[q]   i      ( i    )
		if(i==k)
			return arr[q];
		//  i  k，   i       arr[p..q-1] 
		else if(i<k)
			return randomizedSort(arr,p,q-1,i);
		//  i  k，   i       arr[q+1..r] ，       ， i          i-k      
		else
			return randomizedSort(arr,q+1,r,i-k);
	}
	/**
	 *    arr[p..r]   a[p..q] a[q+1..r]     ，  a[p..q]       a[q]，
	 * a[q+1..r]       a[q]。
	 * @param arr         
	 * @param p          
	 * @param r          
	 */
	public int partition(int arr[],int p,int r){
		//      ，        ，           
		int x = (int) (Math.random()*(r-p)+p);
		//arr[x] arr[r]  
		int temp = arr[x];
		arr[x] = arr[r];
		arr[r] = temp;
		
		//i              ，          ( i+1)              ，    p      
		int i = p-1;
		// p     ，     r       
		for(int j=p;j<r;j++){
			//  arr[j]  arr[r]    ，        ，  i                ，
			//  arr[j]     i+1   
			if(arr[j]<temp){
				i++;
				int w = arr[j];
				arr[j]= arr[i];
				arr[i]= w;
			}
		}
		//   arr[r]               
		arr[r] = arr[i+1];
		arr[i+1] = temp;
		return i+1;
	}
	//  .....
	public static void main(String[] args) {
		int arr[] = {3,2,9,0,7,5,4,8,6,1};
		RandomizedSort rs = new RandomizedSort();
		int result = rs.randomizedSort(arr, 0, arr.length-1, 5);
		System.out.println(result);

	}
}