调了好久,发现两个问题:
首先插入的时候忘记把p赋给fa的孩子,还有就是pushup的时候加上一个判断
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
#define Key_value ch[ch[root][1]][0]
using namespace std;
const int maxn=100010;
const int INF=1000000000;
int N,M;
int res;
int ch[maxn][2],pre[maxn],size[maxn],key[maxn];
int root,tot1,tot2,s[maxn];
int det;
void NewNode(int &r,int f,int val)
{
if(tot2)r=s[tot2--];
else r=++tot1;
ch[r][0]=ch[r][1]=0;
pre[r]=f;
key[r]=val;
size[r]=1;
}
void pushup(int r)
{
if(r)//加了这个就对了
size[r]=size[ch[r][0]]+size[ch[r][1]]+1;
}
void Rotate(int x,int kind)
{
int y=pre[x];
ch[y][!kind]=ch[x][kind];
pre[ch[x][kind]]=y;
if(pre[y])ch[pre[y]][ch[pre[y]][1]==y]=x;
pre[x]=pre[y];
ch[x][kind]=y;
pre[y]=x;
pushup(y);
}
void Splay(int r,int goal)
{
while(pre[r]!=goal)
{
if(pre[pre[r]]==goal)
{
Rotate(r,ch[pre[r]][0]==r);
}
else
{
int y=pre[r];
int kind=(ch[pre[y]][0]==y);
if(ch[y][kind]==r)
{
Rotate(r,!kind);
Rotate(r,kind);
}
else
{
Rotate(y,kind);
Rotate(r,kind);
}
}
}
pushup(r);
if(goal==0)root=r;
}
int get_kth(int r,int k)
{
int t=size[ch[r][0]]+1;
if(t==k)return r;
if(t>k)return get_kth(ch[r][0],k);
return get_kth(ch[r][1],k-t);
}
void debug(int r)
{
if(!r)return;
printf("%d**%d**%d**%d\n",key[r],r,ch[r][0],ch[r][1]);
debug(ch[r][0]);
debug(ch[r][1]);
}
int Insert(int x)
{
int p=root,fa=0;
while(p)
{
if(key[p]>=x)fa=p,p=ch[p][0];
else fa=p,p=ch[p][1];
}
NewNode(p,fa,x);
if(key[fa]<x)ch[fa][1]=p;
else ch[fa][0]=p;
Splay(p,0);
pushup(root);
return p;
}
void Increase(int x)
{
det+=x;
}
void Decrease(int x)
{
det-=x;
int pos=Insert(M-det);
res+=size[ch[root][0]];
root=ch[root][1];
pre[root]=0;
pushup(root);
}
void Find(int k)
{
if(size[root]<k)
{
printf("-1\n");
return;
}
printf("%d\n",key[get_kth(root,size[root]-k+1)]+det);
}
int main()
{
scanf("%d%d",&N,&M);
char op[5];
int k;
det=res=0;
while(N--)
{
scanf("%s%d",op,&k);
if(op[0]=='I'&&k>=M){Insert(k-det);}
else if(op[0]=='A')Increase(k);
else if(op[0]=='S')Decrease(k);
else if(op[0]=='F')Find(k);
}
printf("%d\n",res);
return 0;
}