BZOJ1503: [NOI2004]郁闷的出纳员（洛谷P1486）

最新推荐文章于 2018-11-02 20:55:52 发布

forezxl

最新推荐文章于 2018-11-02 20:55:52 发布

阅读量688

点赞数 1

CC 4.0 BY-SA版权

分类专栏： BZOJ 洛谷数据结构---平衡树蒟蒻zxl的Blog专栏文章标签：平衡树 Treap BZOJ 洛谷

本文链接：https://blog.youkuaiyun.com/a1799342217/article/details/78648293

蒟蒻zxl的Blog专栏同时被 3 个专栏收录

460 篇文章

订阅专栏

洛谷

292 篇文章

订阅专栏

BZOJ

281 篇文章

订阅专栏

该博客介绍了如何使用平衡树，特别是Treap，来解决BZOJ1503和洛谷P1486题目中的工资管理问题。通过转换加减工资的操作为调整工资下限和上限，可以简化平衡树的实现。文章提供了一种利用Treap进行插入、查询和删除操作的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

平衡树

BZOJ题目传送门
 洛谷题目传送门

显然用平衡树维护（然而有大佬用了线段树）。插入、查询、删除平衡树都可以支持，但是这里的加减操作有点不常规，我们需要进行转化一下。

新建一个变量表示当前工资下限，那么加工资就变成了减工资下限，减工资就变成了加工资上限。新建节点时需要把工资加上（当前工资下限-原来工资下限），输出时也要减掉。这样就不需要麻烦地打标记。

这里给出Treap的代码：

#include<cstdio>
#include<cstdlib>
#define N 100000
using namespace std;
struct node{
    int w,p,size,rnd,to[2];
}t[N+5];
int n,m,rt,nd,wg,ans;
void rtt(int &x,int fl){
    int s=t[x].to[fl];
    t[x].to[fl]=t[s].to[fl^1];
    t[s].to[fl^1]=x;
    t[s].size=t[x].size;
    t[x].size=t[t[x].to[0]].size+t[t[x].to[1]].size+t[x].p;
    x=s; return;
}
void nsrt(int &x,int w){
    if (!x){
        t[x=++nd].w=w; t[x].size=t[x].p=1;
        t[x].rnd=rand(); return;
    }
    t[x].size++;
    if (t[x].w==w) t[x].p++;
    else{
        int flag=t[x].w<w;
        nsrt(t[x].to[flag],w);
        if (t[x].rnd>t[t[x].to[flag]].rnd) rtt(x,flag);
    }
}
void dlt(int &x,int w){
    if (!x) return;
    if (t[x].w==w){
        if (t[x].p>1) { t[x].p--; t[x].size--; return; }
        if (!(t[x].to[0]*t[x].to[1])) x=t[x].to[0]+t[x].to[1];
        else if (t[t[x].to[0]].rnd<t[t[x].to[1]].rnd) rtt(x,0),dlt(x,w);
        else rtt(x,1),dlt(x,w);
    }
    else{
        t[x].size--;
        if (t[x].w<w) dlt(t[x].to[1],w);
        else dlt(t[x].to[0],w);
    }
}
int srch_nm(int x,int w){
    if (!x) return -1;
    if (t[t[x].to[0]].size>=w)
        return srch_nm(t[x].to[0],w);
    if (t[t[x].to[0]].size+t[x].p>=w)
        return t[x].w-wg+m;
    return srch_nm(t[x].to[1],w-t[t[x].to[0]].size-t[x].p);
}
int srch_frnt(int x,int w){
    if (!x) return -0x7fffffff;
    if (t[x].w<w){
        int k=srch_frnt(t[x].to[1],w);
        if (k!=-0x7fffffff) 
            return k;
        else return t[x].w;
    }
    else return srch_frnt(t[x].to[0],w);
}
int main(){
    scanf("%d%d",&n,&m); wg=m;
    while (n--){
        int x; char s[5];
        scanf("%s%d",&s,&x);
        switch (s[0]){
            case 'I': if (x+wg-m>=wg) nsrt(rt,x+wg-m); break;
            case 'A': wg-=x; break;
            case 'S': wg+=x; break;
            case 'F': x=t[rt].size-x+1,printf("%d\n",srch_nm(rt,x)); break;
        }
        int p=srch_frnt(rt,wg);
        while (p!=-0x7fffffff) 
            dlt(rt,p),p=srch_frnt(rt,wg),ans++;
    }
    return printf("%d\n",ans),0;
}