【BZOJ 1901】 Zju2112 Dynamic Rankings 主席树+树状数组

说明:提交一直segmentation fault,没有accepted
题意:
带修改的区间第k大查询;

#include<bits/stdc++.h>
using namespace std;
#define lson l,mid
#define rson mid+1,r
const int maxn=60001;
int hush[maxn],a[50001],ls[maxn<<5],rs[maxn<<5],sum[maxn<<5],root[maxn];
int n,m,tot,tota,totb,totn,A[50],B[50];
struct qu
{
    int l,r,k;
} q[10001];
int build(int l,int r)
{
    int rt=++tot;
    sum[rt]=0;
    if(l<r)
    {
        int mid=l+r>>1;
        ls[rt]=build(lson);
        rs[rt]=build(rson);
    }
    return rt;
}
int lowbit(int x)
{
    return x&(-x);
}
int fin(int x)
{
    return lower_bound(hush+1,hush+1+totn,x)-hush;
}
int update(int last,int p,int val,int l,int r)
{
    int rt=++tot;
    sum[rt]=sum[last]+val;
    ls[rt]=ls[last],rs[rt]=rs[last];
    if(l<r)
    {
        int mid=l+r>>1;
        if(p<=mid)
            ls[rt]=update(ls[rt],p,val,lson);
        else
            rs[rt]=update(rs[rt],p,val,rson);
    }
    return rt;
}
void Add(int x,int p,int val)
{
    for(int i=x; i<=n; i+=lowbit(i))
        root[i]=update(root[i],p,val,1,totn);
}
int query(int k,int l,int r)
{
    if(l==r)return l;
    int sum1=0,sum2=0;
    for(int i=1; i<=tota; i++)
        sum1+=sum[A[i]];
    for(int i=1; i<=totb; i++)
        sum2+=sum[B[i]];
    int tmp=sum2-sum1;
    int mid=l+r>>1;
    if(tmp>=k)
    {
        for(int i=1; i<=tota; i++)
            A[i]=ls[A[i]];
        for(int i=1; i<=totb; i++)
            B[i]=ls[B[i]];
        return query(k,lson);
    }
    else
    {
        for(int i=1; i<=tota; i++)
            A[i]=rs[A[i]];
        for(int i=1; i<=totb; i++)
            B[i]=rs[B[i]];
        return query(k-tmp,rson);
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        tot=tota=totb=totn=0;
        scanf("%d%d",&n,&m);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            hush[i]=a[i];
        }
        totn=n;
        char s[3];
        for(int i=1; i<=m; i++)
        {
            scanf("%s",s);
            if(s[0]=='Q')
                scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].k);
            else
            {
                q[i].l=0;
                scanf("%d%d",&q[i].r,&q[i].k);
                hush[++totn]=q[i].k;
            }
        }
        sort(hush+1,hush+totn+1);
        totn=unique(hush+1,hush+1+totn)-hush-1;
        root[0]=build(1,totn);
        for(int i=1;i<=n;i++)
            root[i]=root[0];
        for(int i=1; i<=n; i++)
        {
            Add(i,fin(a[i]),1);
        }
        for(int i=1; i<=m; i++)
        {
            if(q[i].l)
            {
                tota=totb=0;
                for(int j=q[i].l-1; j; j-=lowbit(j))
                    A[++tota]=ls[root[j]];
                for(int j=q[i].r; j; j-=lowbit(j))
                    B[++totb]=ls[root[j]];
                printf("%d\n",query(q[i].k,1,totn));
            }
            else
            {
                int tp=fin(a[q[i].r]);
                Add(q[i].r,tp,-1);
                a[q[i].r]=q[i].k;
                Add(q[i].r,fin(a[q[i].r]),1);
            }
        }
    }
    return 0;
}
发布了174 篇原创文章 · 获赞 64 · 访问量 6万+
展开阅读全文

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

©️2019 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览