bzoj 1901 Zju2112 Dynamic Rankings 树状数组套线段树

题目大意:
维护一个数列,要求单点修改以及求区间第kk大。
n104,m104,ai[0,109]n≤10^4,m≤10^4,a_i\in[0,10^9]

分析:
对每一个位置维护一颗权值线段树,然后外面套树状数组即可。
修改直接在数状数组对应位置的线段树修改。查询同主席树。

代码:

/**************************************************************
    Problem: 1901
    User: ypxrain
    Language: C++
    Result: Accepted
    Time:516 ms
    Memory:49428 kb
****************************************************************/
 
#include <iostream>
#include <cstdio>
#include <cmath>
 
const int maxn=2e4+7;
const int maxa=1e9;
 
using namespace std;
 
int n,m,tot,x,y,k,num;
int a[maxn],root[maxn],p[maxn],ka[maxn];
char op[3];
 
struct node{
    int l,r,data;
}t[maxn*204];
 
void ins(int &p,int l,int r,int x,int k)
{
    if (!p) p=++tot;
    t[p].data+=k;
    if (l==r) return;
    int mid=(l+r)/2;
    if (x<=mid) ins(t[p].l,l,mid,x,k);
           else ins(t[p].r,mid+1,r,x,k); 
}
 
void updata(int num,int x,int k)
{
    for (int i=num;i<=n;i+=i&(-i))
    {
        ins(root[i],0,maxa,x,k);
    }
}
 
int getans(int l,int r,int k)
{
    if (l==r) return l;
    int sum=0;
    for (int i=1;i<=num;i++) sum+=t[t[p[i]].l].data*ka[i];
    int mid=(l+r)/2;
    if (sum>=k)
    {
        for (int i=1;i<=num;i++) p[i]=t[p[i]].l;
        getans(l,mid,k);
    }
    else
    {
        for (int i=1;i<=num;i++) p[i]=t[p[i]].r;
        getans(mid+1,r,k-sum);
    }
}
 
int main()
{
    scanf("%d%d",&n,&m);    
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        updata(i,a[i],1);
    }   
    for (int i=1;i<=m;i++)
    {
        scanf("%s",op);
        if (op[0]=='C')
        {
            scanf("%d%d",&x,&y);
            updata(x,a[x],-1);
            a[x]=y;
            updata(x,a[x],1);
        }
        else
        {
            scanf("%d%d%d",&x,&y,&k);
            num=0;
            for (int i=x-1;i>0;i-=i&(-i)) p[++num]=root[i],ka[num]=-1;
            for (int i=y;i>0;i-=i&(-i)) p[++num]=root[i],ka[num]=1;
            printf("%d\n",getans(0,maxa,k));            
        }
    }
}
发布了720 篇原创文章 · 获赞 37 · 访问量 13万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览