AC日记——[ZJOI2007]报表统计 bzoj 1058

Posted 2020-09-18 Only U - IU

1058

 

思路：

　　平衡树的题；

　　然而我的平衡树写一次炸一次QwQ；

　　而且各种tle；

　　所以stl水过；

 

代码：

#include <set>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define maxn 1000005

class HeadType {
    private:
        int head[maxn*2],cnt;
        
    public:
        bool size()
        {
            return cnt;
        }
        
        int top()
        {
            if(!cnt) return 0x7fffffff;
            return head[1];
        }
        
        inline void push(int x)
        {
            head[++cnt]=x;
            int now=cnt,tmp;
            while(now>1)
            {
                if(head[now>>1]>head[now]) tmp=head[now],head[now]=head[now>>1],now>>=1,head[now]=tmp;
                else break;
            }
        }
        
        inline void pop()
        {
            head[1]=head[cnt--];
            if(cnt==0) return ;
            int pos,pos_,now=1;
            while(1)
            {
                pos=now,pos_=head[now];
                if((now<<1)<=cnt&&head[now<<1]<head[pos]) pos=now<<1,pos_=head[now<<1];
                if((now<<1|1)<=cnt&&head[now<<1|1]<head[pos]) pos=now<<1|1,pos_=head[now<<1|1];
                if(pos==now) return ;
                head[pos]=head[now],head[now]=pos_,now=pos;
            }
        }
};
class HeadType ai,bi,ci;

struct ListType {
    int pre,dis;
};
struct ListType bili[maxn];

int n,m,vi[maxn],tot;

set<int>so;

inline void in(int &now)
{
    int if_z=1;now=0;
    char Cget=getchar();
    while(Cget>‘9‘||Cget<‘0‘)
    {
        if(Cget==‘-‘) if_z=-1;
        Cget=getchar();
    }
    while(Cget>=‘0‘&&Cget<=‘9‘)
    {
        now=now*10+Cget-‘0‘;
        Cget=getchar();
    }
    now*=if_z;
}

int main()
{
    in(n),in(m);bili[0].dis=1e9,bili[n+1].dis=1e9,tot=n+1;
    for(int i=1;i<=n;i++)
    {
        bili[i].pre=i-1,in(bili[i].dis);
        ai.push(abs(bili[i].dis-bili[i-1].dis));
        so.insert(bili[i].dis),vi[i]=bili[i].dis;
    }
    sort(vi+1,vi+n+1),bili[n+1].pre=n,ai.push(abs(bili[n+1].dis-bili[bili[n+1].pre].dis));
    for(int i=2;i<=n;i++) ci.push(abs(vi[i]-vi[i-1]));
    char op[15];int p,x;
    for(;m--;)
    {
        scanf("%s",op);
        if(op[0]==‘I‘)
        {
            in(p),in(x);p++;
            bi.push(abs(bili[p].dis-bili[bili[p].pre].dis));
            ai.push(abs(x-bili[bili[p].pre].dis)),ai.push(abs(bili[p].dis-x));
            bili[++tot].pre=bili[p].pre,bili[tot].dis=x,bili[p].pre=tot;
            set<int>::iterator it=so.lower_bound(x);
            if(it!=so.end()) ci.push(abs(*it-x));
            if(it!=so.begin()) it--,ci.push(abs(*it-x));
            so.insert(x);
        }
        else if(op[4]==‘S‘) printf("%d\n",ci.top());
        else
        {
            while(ai.top()==bi.top()) ai.pop(),bi.pop();
            printf("%d\n",ai.top());
        }
    }
    return 0;
}

 

平衡树代码：

#pragma GCC optimize("O2")
#include <cstdio>
#include <iostream>
#include <algorithm>
 
using namespace std;
 
#define maxn 1000005
#define INF 0x7fffffff
 
class SplayTreeType {
    private:
        int ch[maxn][2],key[maxn],minkey[maxn],size[maxn];
        int pre[maxn],f[maxn];
         
    public:
        int ai[maxn],tot,root,n;
         
        inline void updata(int now)
        {
            size[now]=1,minkey[now]=abs(key[now]-pre[now]);
            if(ch[now][0]) size[now]+=size[ch[now][0]],minkey[now]=min(minkey[now],minkey[ch[now][0]]);
            if(ch[now][1]) size[now]+=size[ch[now][1]],minkey[now]=min(minkey[now],minkey[ch[now][1]]);
        }
         
        inline int tree_build(int l,int r,int fa)
        {
            int now=l+r>>1;
            key[now]=ai[now],pre[now]=ai[now-1],f[now]=fa;
            minkey[now]=abs(key[now]-pre[now]);
            if(now>l) ch[now][0]=tree_build(l,now-1,now);
            if(now<r) ch[now][1]=tree_build(now+1,r,now);
            updata(now);return now;
        }
         
        inline void rotate(int now,int &to)
        {
            int fa=f[now],ffa=f[fa];bool pos=(ch[f[now]][1]==now);
            if(fa==to) to=now;
            else if(ffa) ch[ffa][ch[f[fa]][1]==fa]=now;
            ch[fa][pos]=ch[now][pos^1];
            if(ch[fa][pos]) f[ch[fa][pos]]=fa;
            ch[now][pos^1]=fa,f[fa]=now,f[now]=ffa;
            updata(fa);
        }
         
        inline void splay(int now,int &to)
        {
            while(now!=to)
            {
                int fa=f[now],ffa=f[fa];
                if(fa!=to) rotate((ch[f[now]][1]==now)==(ch[f[fa]][1]==fa)?fa:now,to);
                rotate(now,to);
            }
            updata(now);
        }
         
        inline void insert(int p,int ci)
        {
            splay(p+2,root),pree();
            ch[ch[root][0]][1]=++tot;
            f[tot]=ch[root][0],key[tot]=ci,pre[tot]=key[ch[root][0]];
            pre[root]=key[tot],updata(tot),updata(ch[root][0]),updata(root);
        }
         
        inline void pree()
        {
            int now=ch[root][0];
            while(ch[now][1]) now=ch[now][1];
            splay(now,ch[root][0]);
        }
         
        inline void suff()
        {
            int now=ch[root][1];
            while(ch[now][0]) now=ch[now][0];
            splay(now,ch[root][1]);
        }
         
        inline void insert_(int ci)
        {
            int now=root,fa=0;
            while(1)
            {
                fa=now;
                if(key[now]<=ci) now=ch[now][1];
                else now=ch[now][0];
                if(now==0)
                {
                    now=++tot;
                    ch[fa][ci>key[fa]]=now;
                    f[now]=fa,key[now]=ci,size[now]=1;
                    splay(now,root),pree(),suff();
                    pre[root]=key[ch[root][0]],pre[ch[root][1]]=key[root];
                    updata(ch[root][0]),updata(ch[root][1]),updata(root);return ;
                }
            }
        }
         
        inline int query()
        {
            splay(2,root),splay(n+2,ch[root][1]);
            return minkey[ch[ch[root][1]][0]];
        }
};
struct SplayTreeType ai,bi;
 
int n,m;
 
inline void in(int &now)
{
    int if_z=1;now=0;
    char Cget=getchar();
    while(Cget>‘9‘||Cget<‘0‘)
    {
        if(Cget==‘-‘) if_z=-1;
        Cget=getchar();
    }
    while(Cget>=‘0‘&&Cget<=‘9‘)
    {
        now=now*10+Cget-‘0‘;
        Cget=getchar();
    }
    now*=if_z;
}
 
int main()
{
    in(n),in(m);ai.tot=n+2,bi.tot=n+2,ai.n=n,bi.n=n;
    for(int i=2;i<=n+1;i++) in(ai.ai[i]),bi.ai[i]=ai.ai[i];char op[15];int u,v,x;
    sort(bi.ai+2,bi.ai+n+2);bi.ai[1]=-(INF-1),bi.ai[bi.tot]=INF;
    ai.root=ai.tree_build(1,n+2,0),bi.root=bi.tree_build(1,n+2,0);
    for(;m--;)
    {
        scanf("%s",op);
        if(op[0]==‘I‘) in(u),in(v),ai.insert(u,v),bi.insert_(v);
        else if(op[0]==‘M‘)
        {
            if(op[4]==‘S‘) printf("%d\n",bi.query());
            else printf("%d\n",ai.query());
        }
    }
    return 0;
}