当前位置: 代码迷 >> 综合 >> P1486 [NOI2004] 郁闷的出纳员(splay模板整理)
  详细解决方案

P1486 [NOI2004] 郁闷的出纳员(splay模板整理)

热度:8   发布时间:2024-01-12 15:04:03.0

传送门

splay的优势:支持对区间进行操作:查询区间信息,打区间标记。
更简洁的平衡树可以用multiset。

花了好长时间

#include <bits/stdc++.h>using namespace std;
//-----pre_def----
const double PI = acos(-1.0);
const int INF = 1e9;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
#define fir(i, a, b) for (int i = (a); i <= (b); i++)
#define rif(i, a, b) for (int i = (a); i >= (b); i--)
#define endl '\n'
#define init_h memset(h, -1, sizeof h), idx = 0;
#define lowbit(x) x &(-x)//---------------
const int N = 1e5 + 10;struct node
{
    int s[2], p, v; //s[0]==lson,s[1]==rson,p=pre,v=val(注意:splay中的val不作为排序依据,只在初始状态下有序)int size;       //size==包括自身在内的子树的节点数//int lz;//懒标记,可以记录翻转信息void init(int _v, int _p){
    v = _v;p = _p;size = 1;}
} tr[N];
int n, m, root, idx;
int delta;
//功能函数
void pushup(int u);       //由子向父更新
void rotate(int x);       //核心函数1:将所有情况的左旋右旋写在一个函数里面
void splay(int x, int k); //核心函数2:将x节点旋转至k下面,如果k是0则旋转到根
int find(int v);          //查找v的位置,并将其旋转到根节点
int insert(int v);        //插入v
int find_pre(int v);      //查找v的前驱的节点编号
int find_ne(int v);       //查找v的后继的节点编号
void del(int v);          //删除值为v
int find_k(int k);        //查找排名为k的数
void del(int l, int r);   //删除值为l~r的所有点
int get(int v);
//over
void pushup(int u)
{
    tr[u].size = tr[tr[u].s[0]].size + tr[tr[u].s[1]].size + 1;
}
void rotate(int x) //核心函数1:将所有情况的左旋右旋写在一个函数里面
{
    int y = tr[x].p, z = tr[y].p; //在这个链上:z是祖宗,y是爸爸,x是儿子。现在要把x移动到最上边int k = tr[y].s[1] == x;      // k=0表示x是y的左儿子;k=1表示x是y的右儿子tr[z].s[tr[z].s[1] == y] = x;tr[x].p = z;tr[y].s[k] = tr[x].s[k ^ 1];tr[tr[x].s[k ^ 1]].p = y;tr[x].s[k ^ 1] = y;tr[y].p = x;pushup(y), pushup(x); //顺序不能乱
}
void splay(int x, int k) //核心函数2:将x节点旋转至k下面,如果k是0则旋转到根
{
    while (tr[x].p != k) //但x的父节点不为k时{
    int y = tr[x].p, z = tr[y].p;if (z != k) //x的祖宗还不是k{
    if ((tr[y].s[1] == x) ^ (tr[z].s[1] == y)) //异或,当zyx不为一条直链的时候{
    rotate(x);}else{
    rotate(y);}}rotate(x);}if (!k) //如果k==0,等加于将x旋转至rootroot = x;
}
int find(int v) //查找v的位置,并将其旋转到根节点(此时左子树的size就是v的排名)
{
    int u = root;if (!u)return -1;                               //树空while (tr[u].s[v > tr[u].v] && v != tr[u].v) //当存在儿子并且当前位置的值不等于xu = tr[u].s[v > tr[u].v];                //跳转到儿子,查找x的父节点splay(u, 0);                                 //把当前位置旋转到根节点return root;
}
int insert(int v) //插入一个数,并返回节点编号
{
    int u = root, p = 0;while (u) //找到应该要插入的位置{
    p = u, u = tr[u].s[v > tr[u].v];}u = ++idx;if (p)tr[p].s[v > tr[p].v] = u;tr[u].init(v, p);splay(u, 0);return u;
}
void del(int v) //删除值为v
{
    int last = find_pre(v); //查找x的前驱int next = find_ne(v);  //查找x的后继splay(last, 0);splay(next, last);//将前驱旋转到根节点,后继旋转到根节点下面//很明显,此时后继是前驱的右儿子,x是后继的左儿子,并且x是叶子节点int del = tr[next].s[0]; //后继的左儿子tr[next].s[0] = 0;       //这个节点直接丢掉(不存在了)
}
void del(int l, int r) //删除值为l~r的所有点
{
    int last = get(l); //查找x的前驱int next = get(r); //查找x的后继splay(next, 0);splay(last, next);tr[last].s[1] = 0; //这个节点直接丢掉(不存在了)pushup(last), pushup(next);
}int find_pre(int v) //查找v的前驱的节点编号
{
    find(v);int u = root; //根节点,此时x的父节点(存在的话)就是根节点if (tr[u].v < v)return u;   //如果当前节点的值小于x并且要查找的是前驱u = tr[u].s[0]; //查找后继的话在右儿子上找,前驱在左儿子上找while (tr[u].s[1])u = tr[u].s[1]; //要反着跳转,否则会越来越大(越来越小)return u;           //返回位置
}
int find_ne(int v) //查找v的后继的节点编号
{
    find(v);int u = root; //根节点,此时x的父节点(存在的话)就是根节点if (tr[u].v > v)return u;   //如果当前节点的值小于x并且要查找的是前驱u = tr[u].s[1]; //查找后继的话在右儿子上找,前驱在左儿子上找while (tr[u].s[0])u = tr[u].s[0]; //要反着跳转,否则会越来越大(越来越小)return u;           //返回位置
}
int get(int v)
{
    int u = root, res;while (u){
    if (tr[u].v >= v)res = u, u = tr[u].s[0];elseu = tr[u].s[1];}return res;
}
int find_k(int k) //查找排名为k的数
{
    int u = root;while (u){
    if (tr[tr[u].s[0]].size >= k)u = tr[u].s[0];else if (tr[tr[u].s[0]].size + 1 == k)return tr[u].v;elsek -= tr[tr[u].s[0]].size + 1, u = tr[u].s[1];}return -1;
}void print(int u)
{
    if (tr[u].s[0])print(tr[u].s[0]);printf("%d ", tr[u].v);if (tr[u].s[1])print(tr[u].s[1]);
}int main()
{
    
#ifndef ONLINE_JUDGEfreopen("in.txt", "r", stdin);freopen("out.txt", "w", stdout);int StartTime = clock();
#endifint T = 1, op, x;scanf("%d%d", &n, &m);int L = insert(INF);int R = insert(-INF);int cnt = 0;int tmpp = 0;while (n--){
    char op[2];int k;scanf("%s%d", op, &k);if (*op == 'I'){
    if (k >= m)k -= delta, insert(k), cnt++;}else if (*op == 'A')delta += k;else if (*op == 'S'){
    delta -= k;//delete//find(m - delta);//cout << "!" << m - delta - 1 << endl;// cout << m - delta << endl;// cout << tr[L].v << " " << tr[R].v << endl;del(-INF, m - delta);}else{
    tmpp++;//cout << -tr[root].size << " " << delta << endl;if (tr[root].size - 2 < k)puts("-1");else{
    int tmp = find_k(tr[root].size - k) + delta;printf("%d\n", tmp);}}//cout << -tr[root].size + 2 << " " << delta << endl;}printf("%d\n", cnt - (tr[root].size - 2));
#ifndef ONLINE_JUDGEprintf("Run_Time = %d ms\n", clock() - StartTime);
#endifreturn 0;
}