BZOJ 3110: [Zjoi2013]K大数查询|线段树套线段树

这个题貌似有很多姿势都可以做。
我用的是线段树套线段树。
因为存在区间插入操作,所以我们可以让外层的线段树为权值线段树,内层的线段树为这个权值区间的数在原序列中的位置数量。
这样插入操作就相当于外层的线段树单点修改,内层的区间修改。
然后由于考虑到内存的限制,如果线段树节点全开内存肯定会爆而且会TLE。
这时候我们可以考虑lazy标记,然后有些节点由于父节点有了标记可能直接没开这个节点,这时候我们有两种做法:
1.动态开内存,不下放标记,直接让标记永久化,这样既节省时间又节省内存。
2.动态开内存,下放标记,不让标记永久化,这样既费时又耗内存(不过也可以过!!但是数组千万不要开小喽..
两种写法相差只有几句话。
第一种方法,500W数组就够了实测5+s

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<algorithm>
#include<iostream>
#define N 50050
#define M 7000001
using namespace std;
int sc()
{
    int i=0,f=1; char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i*f;
}
int ch[M][2],sum[M],tag[M];
int root[N*8],n,m,cnt;
void add(int &x,int L,int R,int l,int r)
{
    if(!x)x=++cnt;
    sum[x]+=r-l+1;
    if(L==l&&R==r)
    {
        tag[x]++;
        return;
    }
    int mid=L+R>>1;
    if(r<=mid) add(ch[x][0],L,mid,l,r);
    else if(l>mid) add(ch[x][1],mid+1,R,l,r);
    else add(ch[x][0],L,mid,l,mid),add(ch[x][1],mid+1,R,mid+1,r);
}
void insert(int x,int L,int R,int l,int r,int f)
{
    add(root[x],1,n,l,r);
    if(L==R)return;
    int mid=L+R>>1;
    if(f<=mid)insert(x<<1,L,mid,l,r,f);
    else insert(x<<1|1,mid+1,R,l,r,f);
}
int ask(int x,int L,int R,int l,int r,int a)
{
    if(L==l&&R==r)return sum[x]+(r-l+1)*a;
    int mid=L+R>>1;
    if(r<=mid)return ask(ch[x][0],L,mid,l,r,a+tag[x]);
    else if(l>mid)return ask(ch[x][1],mid+1,R,l,r,a+tag[x]);
    else return ask(ch[x][0],L,mid,l,mid,a+tag[x])+ask(ch[x][1],mid+1,R,mid+1,r,a+tag[x]);
}
int query(int l,int r,int f)
{
    int x=1,L=1,R=n;
    while(L!=R)
    {
        int sum=ask(root[x<<1|1],1,n,l,r,0);
        int mid=L+R>>1;
        if(f<=sum)
            x=x<<1|1,L=mid+1;
        else
            x=x<<1,R=mid,f-=sum;
    }
    return L;
}
int main()
{
    n=sc(),m=sc();
    for(int i=1;i<=m;i++)
    {
        int opt=sc();
        if(opt==1)
        {
            int l=sc(),r=sc(),x=sc();
            insert(1,1,n,l,r,x);
        }
        else
        {
            int l=sc(),r=sc(),x=sc();
            printf("%d\n",query(l,r,x));
        }
    }
    return 0;
}

第二种方法,2000W数组,12+s

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<algorithm>
#include<iostream>
#define N 50050
#define M 20000001
using namespace std;
int sc()
{
    int i=0,f=1; char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i*f;
}
int ch[M][2],sum[M],tag[M];
int root[N*8],n,m,cnt;
void push_down(int x,int L,int R)
{
    if(L==R)return;
    if(!ch[x][0])ch[x][0]=++cnt;
    if(!ch[x][1])ch[x][1]=++cnt;
    int l=ch[x][0],r=ch[x][1],mid=R+L>>1;
    tag[l]+=tag[x];
    tag[r]+=tag[x];
    sum[l]+=tag[x]*(mid-L+1);
    sum[r]+=tag[x]*(R-mid);
    tag[x]=0;
}
void add(int &x,int L,int R,int l,int r)
{
    //if(tag[x])push_down(x,L,R);可有可无 
    if(!x)x=++cnt;
    sum[x]+=r-l+1;
    if(L==l&&R==r)
    {
        tag[x]+=1;
        return;
    }
    int mid=L+R>>1;
    if(r<=mid) add(ch[x][0],L,mid,l,r);
    else if(l>mid) add(ch[x][1],mid+1,R,l,r);
    else add(ch[x][0],L,mid,l,mid),add(ch[x][1],mid+1,R,mid+1,r);
}
void insert(int x,int L,int R,int l,int r,int f)
{
    add(root[x],1,n,l,r);
    if(L==R)return;
    int mid=L+R>>1;
    if(f<=mid)insert(x<<1,L,mid,l,r,f);
    else insert(x<<1|1,mid+1,R,l,r,f);
}
int ask(int x,int L,int R,int l,int r)
{
    if(tag[x])push_down(x,L,R);
    if(L==l&&R==r)return sum[x];
    int mid=L+R>>1;
    if(r<=mid)return ask(ch[x][0],L,mid,l,r);
    else if(l>mid)return ask(ch[x][1],mid+1,R,l,r);
    else return ask(ch[x][0],L,mid,l,mid)+ask(ch[x][1],mid+1,R,mid+1,r);
}
int query(int l,int r,int f)
{
    int x=1,L=1,R=n;
    while(L!=R)
    {
        int sum=ask(root[x<<1|1],1,n,l,r);
        int mid=L+R>>1;
        if(f<=sum)
            x=x<<1|1,L=mid+1;
        else
            x=x<<1,R=mid,f-=sum;
    }
    return L;
}
int main()
{
    n=sc(),m=sc();
    for(int i=1;i<=m;i++)
    {
        int opt=sc();
        if(opt==1)
        {
            int l=sc(),r=sc(),x=sc();
            insert(1,1,n,l,r,x);
        }
        else
        {
            int l=sc(),r=sc(),x=sc();
            printf("%d\n",query(l,r,x));
        }
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值