我TM真是一个弟弟。。。
题意:
给出一串1-N的数字
你每次可以把某个位置的值+1000000
或者找一个值,所有a[1]...a[r]序列的数都不能等于这个值,并且这个值>w
当时比赛觉得肯定是树套树,待修区间第K大,一想不会就自闭了。。。
其实反过来想,如果a[1]....a[r]序列的数都不能等于这个值,那么其实我们可以从a[r+1]....a[n]找到第一个值>=w
但是考虑本题带修改,你会发现这个值加的非常大,大于n,那么一旦某个位置加了这个值,这个值就不再产生贡献
相当于把这个值删掉。
我们考虑把这些值保存起来,因为一旦这个值被删掉,那么意味着,它是有可能作为答案的。
我们建立一颗主席树,并实现查询大于>=w的数的个数,以及区间第K小的操作,并把删除的数放入set里面
我们在区间内部查找>=w的个数,如果这个值为0,我们查询删除的数里面是否有比w大的,如果没有的话,答案就是n+1,否则就是set里面第一个大于w的数,可用lower_bound实现
考虑不为0,我们查第r-l+1+num(大于num的个数)这样就能区间内部查到第一个>=w的数字
再在set里面查询第一个>=w的数字,因为可能1-r区间内被删除的数且这个数比r+1到n区间内>=w的数答案更优秀
两者取最小值即可。
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<set>
using namespace std;
const int maxx = 2e5+6;
struct node{
int l,r;
int cnt;
}tree[maxx*40];
int root[maxx];
int cnt;
set<int>s;
int a[maxx];
void inserts(int l,int r,int pre,int &now,int pos){
now=++cnt;
tree[now]=tree[pre];
tree[now].cnt++;
if(l==r){
return ;
}
int mid=(l+r)>>1;
if (pos<=mid){
inserts(l,mid,tree[pre].l,tree[now].l,pos);
}else{
inserts(mid+1,r,tree[pre].r,tree[now].r,pos);
}
}
int query(int L,int R,int l,int r,int w){
if (l==r){
return tree[R].cnt-tree[L].cnt;
}
int mid=(l+r)>>1;
if (w<=mid){
return tree[tree[R].r].cnt-tree[tree[L].r].cnt+query(tree[L].l,tree[R].l,l,mid,w);
}else {
return query(tree[L].r,tree[R].r,mid+1,r,w);
}
}
int Kth(int L,int R,int l,int r,int k){
if (l==r){
return l;
}
int mid=(l+r)>>1;
int s=tree[tree[R].l].cnt-tree[tree[L].l].cnt;
if (s>=k){
return Kth(tree[L].l,tree[R].l,l,mid,k);
}else {
return Kth(tree[L].r,tree[R].r,mid+1,r,k-s);
}
}
int main(){
int t;
int n,m;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&m);
s.clear();
memset(tree,0,sizeof(tree));
memset(root,0,sizeof(root));
cnt=0;
for (int i=1;i<=n;i++){
scanf("%d",&a[i]);
inserts(1,n,root[i-1],root[i],a[i]);
}
int op;
int r,w,pos=0;
int ans=0;
while(m--){
scanf("%d",&op);
if (op==1){
scanf("%d",&pos);
pos=pos^ans;
s.insert(a[pos]);
}else {
scanf("%d%d",&r,&w);
r=r^ans;
w=w^ans;
int num=query(root[r],root[n],1,n,w);
if (num==0){
auto it=s.lower_bound(w);
if (it!=s.end()){
ans=*it;
printf("%d\n",*it);
}else {
ans=n+1;
printf("%d\n",ans);
}
}else {
auto it=s.lower_bound(w);
if (it!=s.end()){
ans=min(Kth(root[r],root[n],1,n,n-r+1-num),*it);
}else {
ans=Kth(root[r],root[n],1,n,n-r+1-num);
}
printf("%d\n",ans);
}
}
}
}
return 0;
}