一道求一组数的中位数,如果直接排序求解,可能会超时,由于近排刚刚学到树状数组,就想到用这个,有点大才小用,刚开始时定义了全局变量n,又在主函数中定义了一个n,一直求不出来,后来改了还有错,原来发现i=18(2的18次方小于100000000,给成i=20就过了,数据结构真的很好用,今天刚刚看了RMQ和LCA,LCA还有一点不懂,LCA和RMQ转化也不是很清楚,代码如下:
#include<iostream>
using namespace std;
#define maxn 1000010
int c[maxn];
int n;
inline int lowbit(int x)
{
return x&(x^(x-1));
}
void Add(int p,int d)
{
while(p<=maxn)
{
c[p]+=d;
p+=lowbit(p);
}
}
int find(int k)
{
int mod = 0, ans = 0;
for(int i = 20; i >= 0; i --)
{
ans += (1 << i);
if(ans>=maxn||(mod + c[ans] >= k))
ans -= (1 << i);
else
mod += c[ans];
}
return ans + 1;
}
int main()
{
int index,i;
memset(c,0,sizeof(c));
cin>>n;
for(i=0;i<n;i++)
{
scanf("%d",&index);
Add(index,1);
}
cout<<find((n+1)/2)<<endl;
return 0;
}