例题
例一:求不排序数组中的中位数。
要求:
- 不能给数组排序
- 时间复杂度最好在O(n)
分析:
因为题目对时间复杂度和改变数组都做了限制,其实第二条限制已经限制了不能给数组排序,因为如果给数组排序,最快的时间复杂度为NlogN,其中比较快的排序算法就是快速排序。因为我们只需要获得中位数的值,所以也不需要将数组全部排序。
所以,可以采取另一种方法:分治法。 快速排序算法在每一次局部递归后都保证某个元素左侧的值都比它右边的值小,右侧的元素的值都比它大。因此,可以利用这个思路快速的找到第N大的元素,而与快速排序算法不同的是,这种方法关注的并不是元素的左右两边,而仅仅是某一边。
步骤
当使用一次类快速排序算法后,分割元素的下标pos;
- 当pos>length/2,说明中位数在数组左半部分,继续在数组左半部分查找。
- 当pos==length/2时,说明找到该中位数,返回A[pos]即可。
- 当pos<length/2时,说明中位数在数组右半部分,那么继续在数组右半部分找。
注意:
因为数组的长度可能为奇数,也可能为偶数,当数组长度为奇数时,返回的就是数组的中间元素,当数组的长度为偶数时,返回的元素为中间两个元素的平均数。
所以需要注意的就是最后返回时的两种情况,如果数组长度是奇数,那么直接返回中间元素就好;如果是偶数,实际上此时只找到了一个中位数,后半部分的数仍然有可能没有排序,所以这时需要加一个循环,找出右半部分数中最小的数。
代码
public class GetMid1 {
public static void main(String[] args) {
int nums[] = {
1,3,5,9,7,11};
System.out.println(getMid(nums));
}
public static int getMid(int[] nums){
int low = 0;
int high = nums.length-1;
int mid = (low+high)/2;
int pos;
while(true){
pos = partition(nums,low,high);
if(pos == mid)
break;
else if(pos>mid)
high = pos-1;
else
low = pos+1;
}
//如果数组长度为奇数,就直接返回中位数
if(nums.length%2 == 1)
return nums[pos];
//如果数组长度为偶数,需要在找到第二个中位数,则在之后的数中找到最小值
//并和第一个中位数求平均值
else{
int min = getMin(nums,pos+1,nums.length-1