Splay+BZOJ1503

调了好久,发现两个问题:

首先插入的时候忘记把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;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值