分治法解选择问题
2012-05-06 23:25 youxin 阅读(745) 评论(0) 编辑 收藏 举报选择问题: 求n个元素中的第k小的元素。
算法如下:
#include<stdio.h> void swap(int &x,int &y) { int t; t=x;x=y;y=t; } //在a[ left: right]中选择第k小的元素 int select(int a[],int left,int right,int k) { int i,j,pivot; if(left>=right) return a[left]; pivot=a[left]; i=left+1; j=right; while(1) { while(a[i]<pivot) { i=i+1; } while(a[j]>pivot) { j=j-1; } if(i>=j) break; swap(a[i],a[j]); } if(j-left-1+1 +1==k) return pivot; a[left]=a[j]; //存储pivot a[j]=pivot; if(j-left+1<k) return select(a,j+1,right,k-j-1+left); else return select(a,left,j-1,k); } int main() { int a[10]={3,2,5,4,1,6,7,10,9,8}; int res=select(a,0,9,7); printf("%d\n",res); }