即日起在codingBlog上分享您的技术经验即可获得积分,积分可兑换现金哦。

bzoj3110[Zjoi2013]K大数查询 主席树套线段树

编程语言 qq_35866453 12℃ 0评论

题意很简单,就是带修改的K大数查询。


查询自不用说,怎么修改?


由于数字都在1-n范围内,所以他的权值就是他的排名,直接插入,然后建立区间线段树,在线段树内找到修改的范围,然后用主席树区间修改。


因为多打两个else而调了一上午的sb。。。


权值线段树本来也可以,但是数据加强以后被卡了。

#include
#include
#include
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int maxn=1e6+5;
const int M=2e7+5;
typedef long long ll;
int a[maxn],b[maxn],c[maxn];
int N;
bool flag[maxn];
int hash[maxn];
int num[maxn],cnt,op;
int n,m,sz,t,ans,root[maxn],ls[M],rs[M];
ll sum[M],lazy[M];
inline void pushdown(int x,int l,int r,int mid)
{
    if (!lazy[x])return;
    if (!ls[x])ls[x]=++sz;
    sum[ls[x]]+=(mid-l+1)*lazy[x];
    lazy[ls[x]]+=lazy[x];

    if (!rs[x])rs[x]=++sz;
    sum[rs[x]]+=(r-mid)*lazy[x];
    lazy[rs[x]]+=lazy[x];
    lazy[x]=0;
}
inline void update(int last,int &x,int l,int r,int a,int b,ll v)
{
    if (!x)x=++sz;
    int mid=(l+r)>>1;
    ls[x]=ls[last],rs[x]=rs[last];
    if (l>=a&&r<=b)
    {
        sum[x]+=v*(r-l+1);
        lazy[x]+=v;
        return;
    }pushdown(x,l,r,mid);
    if (a<=mid)update(ls[last],ls[x],l,mid,a,b,v);
     if (mid+1<=b)update(rs[last],rs[x],mid+1,r,a,b,v);
    sum[x]=sum[ls[x]]+sum[rs[x]];
}
inline ll query(int x,int l,int r,int a,int b)
{   
    ll ans=0;
    int mid=(l+r)>>1;
    if (a<=l&&r<=b)return sum[x];
    pushdown(x,l,r,mid);
    if (a<=mid)ans+=query(ls[x],l,mid,a,b);
     if (mid+1<=b)ans+=query(rs[x],mid+1,r,a,b);
    return ans;
}

inline void v_update(int x,int l,int r,int v,int a,int b)
{
    int mid=(l+r)>>1;
    update(root[x],root[x],1,n,a,b,1);
    if (l==r)return;
    if (v<=mid)v_update(x<<1,l,mid,v,a,b);
    else v_update(x<<1|1,mid+1,r,v,a,b);
}
inline int v_query(int x,int l,int r,int a,int b,ll k)
{
    if (!x)return 0;
    int mid=(l+r)>>1;
    if (l==r)return l;
    ll t=query(root[x<<1],1,n,a,b);
    if (t>=k) return v_query(x<<1,l,mid,a,b,k);
    else return v_query(x<<1|1,mid+1,r,a,b,k-t);
}
inline int find(int x)
{
    int l=1,r=N,mid;
    while (l<=r)
    {
        mid=(l+r)>>1;
        if (x>hash[mid])l=mid+1;
        else r=mid-1;
    }
    return l;
}
int main()
{
    scanf("%d%d",&n,&m);
    fo(i,1,m)
    {
        scanf("%d%d%d%d",&op,&a[i],&b[i],&c[i]);
        if (op==1)
        {
            flag[i]=true;
            num[++cnt]=c[i];
        }
    }
    sort(num+1,num+1+cnt);
    fo(i,1,cnt)
    {
        if (num[i]!=num[i-1])hash[++N]=num[i];
    } 
    fo(i,1,m)
    {
        if (flag[i])
        {
            t=find(c[i]);
            v_update(1,1,N,t,a[i],b[i]);
        }
        else
        {
            ll tot=query(root[1],1,n,a[i],b[i]);
            ll t=(ll)tot-(ll)c[i]+1;
            ans=v_query(1,1,N,a[i],b[i],t);
            printf("%d\n",hash[ans]);
        }
    }
}

转载请注明:CodingBlog » bzoj3110[Zjoi2013]K大数查询 主席树套线段树

喜欢 (0)or分享 (0)
发表我的评论
取消评论

*

表情