KPI(HDU—5249)权值线段树

题目传送门

Problem Description

你工作以后, KPI 就是你的全部了. 我开发了一个服务,取得了很大的知名度。数十亿的请求被推到一个大管道后同时服务从管头拉取请求。让我们来定义每个请求都有一个重要值。我的KPI是由当前管道内请求的重要值的中间值来计算。现在给你服务记录,有时我想知道当前管道内请求的重要值得中间值。

Input

有大约100组数据。
每组数据第一行有一个n(1≤n≤10000),代表服务记录数。
接下来有n行,每一行有3种形式
“in x”: 代表重要值为x(0≤x≤109)的请求被推进管道。
“out”: 代表服务拉取了管道头部的请求。
"query: 代表我想知道当前管道内请求重要值的中间值. 那就是说,如果当前管道内有m条请求, 我想知道,升序排序后第floor(m/2)+1th 条请求的重要值.
为了让题目简单,所有的x都不同,并且如果管道内没有值,就不会有"out"和"query"操作。

Output

对于每组数据,先输出一行
Case #i:
然后每一次"query",输出当前管道内重要值的中间值。

Sample Input

6
in 874
query
out
in 24622
in 12194
query

Sample Output

Case #1:
874
24622

这道题的解决方法也有好几种,我今天来说一下用权值线段树来解决这道题的方法。
咋一看,就是权值线段树的板子题,写着写着突然发现,这个数的值有点大,我们都知道,权值线段树是用值域来开数组范围大小。所以我们还需要对这个存储进行离散化一下。因为这道题给的操作不多,所以我们可以先储存,然后再挂到树上,这样就可以顺利解决这道题了。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<queue>
using namespace std;
int tree[40005];
int a[10005];
int b[10005];
char s[20];
void build(int l,int r,int k)
{
    if(l==r)
    {
        tree[k]=0;
        return;
    }
    int mid=(l+r)/2;
    build(l,mid,k*2);
    build(mid+1,r,k*2+1);
    tree[k]=tree[k*2]+tree[k*2+1];
}
void add(int l,int r,int k,int x,int op)
{

    if(l==r)
    {
        tree[k]+=op;
        return;
    }
    int mid=(l+r)/2;
    if(mid>=x)
    {
        add(l,mid,k*2,x,op);
    }
    else
    {
        add(mid+1,r,k*2+1,x,op);
    }
    tree[k]=tree[k*2]+tree[k*2+1];
}
int xth(int l,int r,int k,int x)
{
    if(l==r)
    {
        //printf("%d\n",l);
        return l;
    }

    int mid=(l+r)/2;
    if(tree[k*2]>=x)
    {
        return xth(l,mid,k*2,x);
    }
    else
    { //printf("%d %d %d**\n",l,r,tree[2*k]);
        return xth(mid+1,r,k*2+1,x-tree[k*2]);
    }
}
int main()
{
    int n;
    int tot=1;
    while(~scanf("%d",&n))
    {
        queue<int>q;
        int sum=0;
        for(int i=1; i<=n; i++)
        {
            scanf("%s",s);
            if(s[0]=='i')
            {
                scanf("%d",&a[i]);
                b[++sum]=a[i];
            }
            else if(s[0]=='o')
            {
                a[i]=-1;
            }
            else
            {
                a[i]=-2;
            }
        }
       //printf("%d\n",sum);
        build(1,sum,1);
        sort(b+1,b+sum+1);
        printf("Case #%d:\n",tot++);
        for(int i=1;i<=n;i++)
        {
            if(a[i]>=0)
            {
                q.push(a[i]);
                int k=lower_bound(b+1,b+sum+1,a[i])-b;
                add(1,sum,1,k,1);
            }
            else if(a[i]==-1)
            {
                int tem=q.front();

                int k=lower_bound(b+1,b+sum+1,tem)-b;
                add(1,sum,1,k,-1);
                 q.pop();
            }
            else
            {
                //printf("%d##\n",tree[3]);
                printf("%d\n",b[xth(1,sum,1,q.size()/2+1)]);
            }
        }
    }
    return 0;
}
发布了42 篇原创文章 · 获赞 5 · 访问量 910

猜你喜欢

转载自blog.csdn.net/qq_43402296/article/details/104220661