HDU - 4417 Super Mario (区间询问rank)

BanFcc

于 2017-07-18 21:04:02 发布

阅读量261

点赞数

分类专栏：主席树

本文链接：https://blog.youkuaiyun.com/BanFcc/article/details/75330679

版权

主席树专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一个名为'Super Mario'的算法问题，涉及区间查询和数据结构的应用。作者通过C++代码实现了一个解决方案，利用平衡树结构进行区间更新和查询，解答了给定整数数组在经过区间修改后，查询特定位置的rank信息。代码中包括插入、查询和排序等操作，以处理多组测试用例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<bits/stdc++.h>
using namespace std;
const int MAXN=(int)1e5+10;
struct node{
    int l,r,sum;
}T[MAXN*40];
int root[MAXN],a[MAXN],cnt;
vector<int>v;
void insert(int l,int r,int &x,int y,int pos){
    x=++cnt;T[x]=T[y];T[x].sum++;
    if(l==r)return ;
    int mid=l+r>>1;
    if(mid>=pos)insert(l,mid,T[x].l,T[y].l,pos);
    else insert(mid+1,r,T[x].r,T[y].r,pos);
}
int query(int l,int r,int x,int y,int vl){
    if(l==r)return T[y].sum-T[x].sum;
    int mid=l+r>>1;
    if(mid>=vl)return query(l,mid,T[x].l,T[y].l,vl);
    else return T[T[y].l].sum-T[T[x].l].sum+query(mid+1,r,T[x].r,T[y].r,vl);
}
int main()
{
    int t;
    scanf("%d",&t);
    T[0].sum=T[0].l=T[0].r=0;
    for(int ca=1;ca<=t;ca++){
        v.clear();
        cnt=0;
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            v.push_back(a[i]);
        }
        sort(v.begin(),v.end());
        v.erase(unique(v.begin(),v.end()),v.end());
        int vsz=v.size();
        for(int i=1;i<=n;i++)insert(1,vsz,root[i],root[i-1],lower_bound(v.begin(),v.end(),a[i])-v.begin()+1);    
        printf("Case %d:\n",ca);
        while(m--){
            int l,r,x;
            scanf("%d%d%d",&l,&r,&x);
            l++,r++;
            x=upper_bound(v.begin(),v.end(),x)-v.begin();
            if(x==0)printf("0\n");
            else printf("%d\n",query(1,vsz,root[l-1],root[r],x));
        }
    }
    return 0;
}