【题目描述】
您需要写一种数据结构（可参考题目标题），来维护一个有序数列，其中需要提供以下操作：
1.查询k在区间内的排名
2.查询区间内排名为k的值
3.修改某一位值上的数值
4.查询k在区间内的前驱(前驱定义为小于x，且最大的数)
5.查询k在区间内的后继(后继定义为大于x，且最小的数)
【输入格式】
第一行两个数 n,m 表示长度为n的有序序列和m个操作
第二行有n个数，表示有序序列
下面有m行，opt表示操作标号
若opt=1 则为操作1，之后有三个数l,r,k 表示查询k在区间[l,r]的排名
若opt=2 则为操作2，之后有三个数l,r,k 表示查询区间[l,r]内排名为k的数
若opt=3 则为操作3，之后有两个数pos,k 表示将pos位置的数修改为k
若opt=4 则为操作4，之后有三个数l,r,k 表示查询区间[l,r]内k的前驱
若opt=5 则为操作5，之后有三个数l,r,k 表示查询区间[l,r]内k的后继
【输出格式】
对于操作1,2,4,5各输出一行，表示查询结果
【样例输入】
9 6 4 2 2 1 9 4 0 1 1 2 1 4 3 3 4 10 2 1 4 3 1 2 5 9 4 3 9 5 5 2 8 5
【样例输出】
2 4 3 4 9
【提示】
n,m<=50000 保证有序序列所有值在任何时刻满足[0,10^8]但是询问的数未必
【来源】
tyvj1730
树状数组套主席树
查询前驱时 对于当前所到的线段树区间 左半边一定要去 右半边如果查询的那个数!=mid+1对应的数 并且右半边有数 就去
到达叶子节点时 看看叶子节点是否有数 有了就更新答案
后继同理
代码：
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=200000+10;
int root[maxn],ls[maxn*100],rs[maxn*100];
int hash[maxn],size[maxn*100];
int sz;
int tot;
int tt;
int L[100],R[100];
int ll,rr;
int A[maxn],B[maxn];
struct Q{
int p,l,r,k;
}q[maxn];
int qans;
inline int bs(int x){
int l=1,r=tot;
while(l+1<r){
int mid=(l+r)>>1;
if(hash[mid]>=x)
r=mid;
else l=mid+1;
}
if(hash[l]==x)
return l;
else return r;
}
inline int lowbit(int x){
return x&(-x);
}
inline void update(int last,int l,int r,int &rt,int w,int x){
rt=++sz;
size[rt]=size[last]+x;
ls[rt]=ls[last];
rs[rt]=rs[last];
if(l==r)
return;
int mid=(l+r)>>1;
if(w<=mid)
update(ls[last],l,mid,ls[rt],w,x);
else update(rs[last],mid+1,r,rs[rt],w,x);
}
inline int query1(int x,int l,int r){
int a[50],b[50];
if(l>=1&&r<x){
int suml=0,sumr=0;
for(int i=1;i<=ll;i++)
suml+=size[L[i]];
for(int i=1;i<=rr;i++)
sumr+=size[R[i]];
return sumr-suml;
}
else if(l!=r){
for(int i=1;i<=ll;i++)
a[i]=L[i];
for(int i=1;i<=rr;i++)
b[i]=R[i];
int mid=(l+r)>>1,ans=0;
if(mid>=1){
for(int i=1;i<=ll;i++)
L[i]=ls[L[i]];
for(int i=1;i<=rr;i++)
R[i]=ls[R[i]];
ans+=query1(x,l,mid);
}
for(int i=1;i<=ll;i++)
L[i]=a[i];
for(int i=1;i<=rr;i++)
R[i]=b[i];
if(mid<x){
for(int i=1;i<=ll;i++)
L[i]=rs[L[i]];
for(int i=1;i<=rr;i++)
R[i]=rs[R[i]];
ans+=query1(x,mid+1,r);
}
return ans;
}
return 0;
}
inline int query2(int x,int l,int r){
if(l==r)
return l;
int suml=0,sumr=0;
for(int i=1;i<=ll;i++)
suml+=size[ls[L[i]]];
for(int i=1;i<=rr;i++)
sumr+=size[ls[R[i]]];
int mid=(l+r)>>1;
if(sumr-suml>=x){
for(int i=1;i<=ll;i++)
L[i]=ls[L[i]];
for(int i=1;i<=rr;i++)
R[i]=ls[R[i]];
return query2(x,l,mid);
}
else {
for(int i=1;i<=ll;i++)
L[i]=rs[L[i]];
for(int i=1;i<=rr;i++)
R[i]=rs[R[i]];
return query2(x-(sumr-suml),mid+1,r);
}
}
/*int query2(int k,int l,int r)
{
if(l==r) return l;
int i,suml=0,sumr=0;
for(i=1;i<=ll;i++) suml+=size[ls[L[i]]];
for(i=1;i<=rr;i++) sumr+=size[ls[R[i]]];
int mid=(l+r)>>1;
if(sumr-suml>=k)
{
for(i=1;i<=ll;i++) L[i]=ls[L[i]];
for(i=1;i<=rr;i++) R[i]=ls[R[i]];
return query2(l,mid,k);
}
else
{
for(i=1;i<=ll;i++) L[i]=rs[L[i]];
for(i=1;i<=rr;i++) R[i]=rs[R[i]];
return query2(mid+1,r,k-(sumr-suml));
}
}*/
inline void query3(int x,int l,int r){
int a[50],b[50];
if(r<x&&l==r){
int suml=0,sumr=0;
for(int i=1;i<=ll;i++)
suml+=size[L[i]];
for(int i=1;i<=rr;i++)
sumr+=size[R[i]];
if(sumr-suml)
qans=max(qans,l);
return ;
}
if(r<x){
int mid=(l+r)>>1;
int suml=0,sumr=0;
for(int i=1;i<=ll;i++)
suml+=size[rs[L[i]]];
for(int i=1;i<=rr;i++)
sumr+=size[rs[R[i]]];
for(int i=1;i<=ll;i++)
a[i]=L[i];
for(int i=1;i<=rr;i++)
b[i]=R[i];
if(sumr-suml){
for(int i=1;i<=ll;i++)
L[i]=rs[L[i]];
for(int i=1;i<=rr;i++)
R[i]=rs[R[i]];
query3(x,mid+1,r);
for(int i=1;i<=ll;i++)
L[i]=a[i];
for(int i=1;i<=rr;i++)
R[i]=b[i];
}
else {
for(int i=1;i<=ll;i++)
L[i]=ls[L[i]];
for(int i=1;i<=rr;i++)
R[i]=ls[R[i]];
query3(x,l,mid);
}
}
else {
int mid=(l+r)>>1;
int suml=0,sumr=0;
for(int i=1;i<=ll;i++)
suml+=size[rs[L[i]]];
for(int i=1;i<=rr;i++)
sumr+=size[rs[R[i]]];
for(int i=1;i<=ll;i++)
a[i]=L[i];
for(int i=1;i<=rr;i++)
b[i]=R[i];
for(int i=1;i<=ll;i++)
L[i]=ls[L[i]];
for(int i=1;i<=rr;i++)
R[i]=ls[R[i]];
query3(x,l,mid);
for(int i=1;i<=ll;i++)
L[i]=a[i];
for(int i=1;i<=rr;i++)
R[i]=b[i];
if(sumr-suml&&mid+1<x){
for(int i=1;i<=ll;i++)
L[i]=rs[L[i]];
for(int i=1;i<=rr;i++)
R[i]=rs[R[i]];
query3(x,mid+1,r);
}
}
}
inline void query4(int x,int l,int r){
int a[50],b[50];
if(l>x&&l==r){
int suml=0,sumr=0;
for(int i=1;i<=ll;i++)
suml+=size[L[i]];
for(int i=1;i<=rr;i++)
sumr+=size[R[i]];
if(sumr-suml)
qans=min(qans,l);
return ;
}
else if(l>x){
int mid=(l+r)>>1;
int suml=0,sumr=0;
for(int i=1;i<=ll;i++)
suml+=size[ls[L[i]]];
for(int i=1;i<=rr;i++)
sumr+=size[ls[R[i]]];
for(int i=1;i<=ll;i++)
a[i]=L[i];
for(int i=1;i<=rr;i++)
b[i]=R[i];
if(sumr-suml){
for(int i=1;i<=ll;i++)
L[i]=ls[L[i]];
for(int i=1;i<=rr;i++)
R[i]=ls[R[i]];
query4(x,l,mid);
for(int i=1;i<=ll;i++)
L[i]=a[i];
for(int i=1;i<=rr;i++)
R[i]=b[i];
}
else {
for(int i=1;i<=ll;i++)
L[i]=rs[L[i]];
for(int i=1;i<=rr;i++)
R[i]=rs[R[i]];
query4(x,mid+1,r);
}
}
else {
int mid=(l+r)>>1;
int suml=0,sumr=0;
for(int i=1;i<=ll;i++)
suml+=size[ls[L[i]]];
for(int i=1;i<=rr;i++)
sumr+=size[ls[R[i]]];
for(int i=1;i<=ll;i++)
a[i]=L[i];
for(int i=1;i<=rr;i++)
b[i]=R[i];
for(int i=1;i<=ll;i++)
L[i]=rs[L[i]];
for(int i=1;i<=rr;i++)
R[i]=rs[R[i]];
query4(x,mid+1,r);
for(int i=1;i<=ll;i++)
L[i]=a[i];
for(int i=1;i<=rr;i++)
R[i]=b[i];
if(sumr-suml&&mid>x){
for(int i=1;i<=ll;i++)
L[i]=ls[L[i]];
for(int i=1;i<=rr;i++)
R[i]=ls[R[i]];
query4(x,l,mid);
}
}
}
int main(){
freopen("psh.in","r",stdin);
freopen("psh.out","w",stdout);
int n,m;
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&A[i]);
B[++tt]=A[i];
}
for(int i=1;i<=m;i++){
scanf("%d",&q[i].p);
if(q[i].p!=3){
scanf("%d %d %d",&q[i].l,&q[i].r,&q[i].k);
if(q[i].p==4||q[i].p==5)
B[++tt]=q[i].k;
}
else {
scanf("%d %d",&q[i].l,&q[i].k);
B[++tt]=q[i].k;
}
}
sort(B+1,B+tt+1);
hash[++tot]=B[1];
for(int i=2;i<=tt;i++)
if(B[i]!=B[i-1])
hash[++tot]=B[i];
for(int i=1;i<=n;i++){
int u=bs(A[i]);
for(int j=i;j<=n;j+=lowbit(j))
update(root[j],1,tot,root[j],u,1);
}
for(int i=1;i<=m;i++){
if(q[i].p==1){
int u=bs(q[i].k);
ll=rr=0;
for(int j=q[i].l-1;j>=1;j-=lowbit(j))
L[++ll]=root[j];
for(int j=q[i].r;j>=1;j-=lowbit(j))
R[++rr]=root[j];
printf("%d\n",query1(u,1,tot)+1);
}
else if(q[i].p==2){
ll=rr=0;
for(int j=q[i].l-1;j>=1;j-=lowbit(j))
L[++ll]=root[j];
for(int j=q[i].r;j>=1;j-=lowbit(j))
R[++rr]=root[j];
printf("%d\n",hash[query2(q[i].k,1,tot)]);
}
else if(q[i].p==3){
int u=bs(A[q[i].l]);
for(int j=q[i].l;j<=n;j+=lowbit(j))
update(root[j],1,tot,root[j],u,-1);
A[q[i].l]=q[i].k;
u=bs(q[i].k);
for(int j=q[i].l;j<=n;j+=lowbit(j))
update(root[j],1,tot,root[j],u,1);
}
else if(q[i].p==4){
int u=bs(q[i].k);
ll=rr=0;
qans=-0x7fffffff;
for(int j=q[i].l-1;j>=1;j-=lowbit(j))
L[++ll]=root[j];
for(int j=q[i].r;j>=1;j-=lowbit(j))
R[++rr]=root[j];
query3(u,1,tot);
printf("%d\n",hash[qans]);
}
else {
int u=bs(q[i].k);
ll=rr=0;
qans=0x7fffffff;
for(int j=q[i].l-1;j>=1;j-=lowbit(j))
L[++ll]=root[j];
for(int j=q[i].r;j>=1;j-=lowbit(j))
R[++rr]=root[j];
query4(u,1,tot);
printf("%d\n",hash[qans]);
}
}
return 0;
}
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
