您的位置:首页 > 其它

2388 -- Who's in the Middle(解题报告)

2011-07-08 11:04 411 查看
http://poj.org/problem?id=2388

问题是要从未经过排序的数组中找到中位数。很多人说这题是水题。其实此题不水。只不过测试的数据水了点。应该让排序的代码都超时,就没人说水题了。

这题的思想是用快排划分的思想。即数组中以一个数字为分水岭,如果大于该数的数字和小于该数的数字数量相等,则该数就是我们要求的中位数。这样算法的时间复杂度是O(n),最差为O(n^2),通过排序的话快排是O(n*lgn),通过小根堆对前一半的数据排序的话是O(n/2*lgn)。相比之下,快排划分的算法优缺点都可以看出来了。

当然随着数据量的增加,当主存无法放下全部的数据的时候,这时应该选择用小根堆排序。因为这时必然要用到外存,而外存的io速度。。。你懂得。

这时使用小根堆只需要扫描数据一遍。(听说面试时,很多人问)

#include<stdio.h>
#include<string.h>
#include<iostream>

void swap(int* arr, int x,int y)
{
int temp = arr[x];
arr[x] = arr[y];
arr[y] = temp;
}
int partition(int* arr, int begin, int end)
{
int i, j;
i = begin;
long pivot = arr[begin];
for(j=begin+1; j<=end; j++)
if(arr[j] <= pivot)
swap(arr, ++i, j);
swap(arr, i, begin);
return i;
}

int select(int* arr, int begin, int end, int i)
{
if(begin == end)
return arr[begin];
int p = partition(arr, begin, end);
int k = p - begin + 1;
if( k == i )
return  arr[p];
else if(k < i)
return select( arr, p+1, end, i - k);
else
return select( arr, begin , p - 1, i);
}

int main()
{
int n;
std::cin>>n;
int q = 0;
int* arr = new int
;
for(int i = 0; i < n; i++)
std::cin>>arr[i];
std::cout<<select(arr, 0, n - 1, (n + 1)/2  )<<std::endl;
delete []arr;
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: