DSOJ Kth Number(第k大的数)

来源:互联网 发布:centos 指定ip访问ssh 编辑:程序博客网 时间:2024/05/01 23:29

题目链接

#include<stdio.h>//查找数组中第k小的元素#define MAX 100/*算法思想:模仿快排的做法,首先去数组中任何一个元素(不妨取第一个),将数组分为两部分,s[0,..p-1],和s[p+1,...n-1],其中左半部分的值都不大于s[p],右半部分的值都不小于s[p]此时如果p==k-1,则说明s[p]即为所要找的第k小的元素,返回s[p]否则,如果p>k-1,说明第k小的元素一定在s[0,...p-1]中则再在s[0,....,p-1]中查找否则,说明第k小的元素在s[p+1,...,n-1]中,而且是s[p+1,...,n-1]中第k-p-1小的元素综上,利用递归即可实现*/int FindKth(int *Array, int n, int k){int left, right, pivot;left = 0;right = n - 1;pivot = Array[0];//选取第一个数组元素作为轴值while (left < right){while (left < right&&Array[right] >= pivot)right--;Array[left] = Array[right];while (left < right&&Array[left] <= pivot)left++;Array[right] = Array[left];}Array[left] = pivot;//left即为轴值所在的位置if (left == k - 1)return Array[left];else if (left>k - 1)return FindKth(Array, left, k);elsereturn FindKth(Array + left + 1, n - left - 1, k - left - 1);}int main(){int Array[MAX];int k, n, i;scanf("%d", &n);for (i = 0; i < n; i++)scanf("%d", Array + i);scanf("%d", &k);printf("%d\n", FindKth(Array, n, k));return 0;}


0 0
原创粉丝点击