HDU ACM 5249 KPI->树状数组+二分

分析:注意队列中保存的是排序后数组中的下标。树状数组也是根据下标进行处理。

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;

#define    N 10005
int c[N<<2];
int n,a[N],b[N];
char str[N][10];

int lowbit(int x)
{
    return x&(-x);
}

void update(int x,int val)
{
    while(x<N)
    {
        c[x]+=val;
        x+=lowbit(x);
    }
}

int sum(int x)
{
    int s=0;

    while(x>0)
    {
        s+=c[x];
        x-=lowbit(x);
    }
    return s;
}

int solve(int x)
{
    int id,l,r,mid;

    id=x/2+1;
    l=1;
    r=N;
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(sum(mid)>=id)
            r=mid-1;
        else
            l=mid+1;
    }
    return b[l];
}

int main()      
{
    int x,T=0,i,cnt;

    while(scanf("%d",&n)==1)
    {
        queue<int> q;
        memset(c,0,sizeof(c));
        cnt=1;
        for(i=1;i<=n;i++)
        {
            scanf("%s",str[i]);
            if(str[i][0]=='i')
            {
                scanf("%d",&x);
                b[cnt++]=x;
                a[i]=x;
            }
        }
        sort(b+1,b+cnt+1);                   //排序
        cnt=unique(b+1,b+cnt+1)-(b+1);         //去重

        printf("Case #%d:\n",++T);
        for(i=1;i<=n;i++)
        {
            if(str[i][0]=='i')
            {
                x=lower_bound(b+1,b+cnt+1,a[i])-b;
                update(x,1);
                q.push(x);           //队列中保存的是下标
            }
            else if(str[i][0]=='o')
            {
                x=lower_bound(b+1,b+cnt+1,b[q.front()])-b;
                update(x,-1);
                q.pop();
            }
            else if(str[i][0]=='q')
                printf("%d\n",solve(q.size()));
        }
    }
    return 0;      
}


郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。