bzoj3110【ZJOI2013】K大数查询

3110: [Zjoi2013]K大数查询

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 3208  Solved: 1361
[Submit][Status][Discuss]

Description

有N个位置，M个操作。操作有两种，每次操作如果是1 a b c的形式表示在第a个位置到第b个位置，每个位置加入一个数c
如果是2 a b c形式，表示询问从第a个位置到第b个位置，第C大的数是多少。

Input

第一行N，M
接下来M行，每行形如1 a b c或2 a b c

Output

输出每个询问的结果

Sample Input

2 5
1 1 2 1
1 1 2 2
2 1 1 2
2 1 1 1
2 1 2 3

Sample Output

1
2
1

HINT

【样例说明】

第一个操作 后位置 1 的数只有 1 ， 位置 2 的数也只有 1 。 第二个操作 后位置 1

的数有 1 、 2 ，位置 2 的数也有 1 、 2 。 第三次询问 位置 1 到位置 1 第 2 大的数 是

1 。 第四次询问 位置 1 到位置 1 第 1 大的数是 2 。 第五次询问 位置 1 到位置 2 第 3

大的数是 1 。‍

N,M<=50000,N,M<=50000

a<=b<=N

1操作中abs(c)<=N

2操作中abs(c)<=Maxlongint

Source

线段树套线段树+二分答案

树套树竟然一次就写对了，那叫一个开心啊！

外层是权值线段树，记录权值为[l,r]的信息。

内层是区间线段树，记录当前权值的位置在[l,r]的数量。

每次加点，把外层所有覆盖该权值的线段树全部更新。

每次查询，二分答案即可。

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<cmath>#include<algorithm>#define F(i,j,n) for(int i=j;i<=n;i++)#define D(i,j,n) for(int i=j;i>=n;i--)#define ll long long#define maxn 20000005using namespace std;int n,m,f,a,b,c,tot;int rt[maxn],ls[maxn],rs[maxn],sum[maxn],tag[maxn];inline int read(){int x=0,f=1;char ch=getchar();while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}inline void update(int k,int l,int r,int x){tag[k]+=x;sum[k]+=(r-l+1)*x;}inline void pushdown(int k,int l,int r){if (!tag[k]||l==r) return;if (!ls[k]) ls[k]=++tot;if (!rs[k]) rs[k]=++tot;int mid=(l+r)>>1;update(ls[k],l,mid,tag[k]);update(rs[k],mid+1,r,tag[k]);tag[k]=0;}inline void pushup(int k){sum[k]=sum[ls[k]]+sum[rs[k]];}inline void add(int &k,int l,int r,int a,int b){if (!k) k=++tot;if (l==a&&r==b){update(k,l,r,1);return;}pushdown(k,l,r);int mid=(l+r)>>1;if (b<=mid) add(ls[k],l,mid,a,b);else if (a>mid) add(rs[k],mid+1,r,a,b);else add(ls[k],l,mid,a,mid),add(rs[k],mid+1,r,mid+1,b);pushup(k);}inline void insert(){int k=1,l=1,r=n;while (l!=r){int mid=(l+r)>>1;add(rt[k],1,n,a,b);if (c<=mid) r=mid,k=k<<1;else l=mid+1,k=k<<1|1;}add(rt[k],1,n,a,b);}inline int query(int k,int l,int r,int a,int b){if (!k) return 0;if (l==a&&r==b) return sum[k];pushdown(k,l,r);int mid=(l+r)>>1;if (b<=mid) return query(ls[k],l,mid,a,b);else if (a>mid) return query(rs[k],mid+1,r,a,b);else return query(ls[k],l,mid,a,mid)+query(rs[k],mid+1,r,mid+1,b);}inline int solve(){int k=1,l=1,r=n;while (l!=r){int mid=(l+r)>>1;int tmp=query(rt[k<<1|1],1,n,a,b);if (tmp>=c) l=mid+1,k=k<<1|1;else c-=tmp,r=mid,k=k<<1;}return l;}int main(){n=read();m=read();while (m--){f=read();a=read();b=read();c=read();if (f==1) insert();else printf("%d\n",solve());}}