spoj 4487. Can you answer these queries VI(伸展树)

题意:给出n个数,有四种操作:①在x前插入一个数②删除x③将x的值替换为y④查询区间[x,y]的最大连续子段和。

思路:伸展树搞,思路不难想……这题写得十分蛋疼,前面小bug改完了各种TLE,加了各种优化才过。。。


代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<stack>
//#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=200000+10;
int num[maxn],n,q;
struct Splay_Tree
{
int ch[maxn][2],pre[maxn],size[maxn],val[maxn];
int pmax[maxn],fmax[maxn],smax[maxn],sum[maxn];
int tot,root;
void Newnode(int & rt,int fa,int v)
{
    rt=++tot;
    ch[rt][0]=ch[rt][1]=0;
    pre[rt]=fa;size[rt]=1;
    val[rt]=pmax[rt]=fmax[rt]=smax[rt]=sum[rt]=v;
}
inline void PushUp(int rt)
{
    int ls=ch[rt][0],rs=ch[rt][1];
    size[rt]=size[ls]+size[rs]+1;
    sum[rt]=sum[ls]+sum[rs]+val[rt];

    smax[rt]=max(smax[ls],smax[rs]);
    smax[rt]=max(smax[rt],val[rt]);
    smax[rt]=max(smax[rt],fmax[ls]+val[rt]);
    smax[rt]=max(smax[rt],val[rt]+pmax[rs]);
    smax[rt]=max(smax[rt],fmax[ls]+val[rt]+pmax[rs]);

    pmax[rt]=max(pmax[ls],sum[ls]+val[rt]);
    pmax[rt]=max(pmax[rt],sum[ls]+val[rt]+pmax[rs]);

    fmax[rt]=max(fmax[rs],sum[rs]+val[rt]);
    fmax[rt]=max(fmax[rt],sum[rs]+val[rt]+fmax[ls]);
}
void Rotate(int x,int kind)
{
    int y=pre[x];
    ch[y][kind^1]=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 rt,int goal)
{
    while(pre[rt]!=goal)
    {
        int y=pre[rt];
        if(pre[y]==goal)
            Rotate(rt,ch[y][0]==rt);
        else
        {
            int kind=ch[pre[y]][0]==y;
            if(ch[y][kind]==rt)
            {
                Rotate(rt,kind^1);
                Rotate(rt,kind);
            }
            else
            {
                Rotate(y,kind);
                Rotate(rt,kind);
            }
        }
    }
    PushUp(rt);
    if(goal==0) root=rt;
}
void RotateTo(int k,int goal)
{
    int rt=root;
    while(size[ch[rt][0]]!=k)
    {
        if(size[ch[rt][0]]>k)
            rt=ch[rt][0];
        else
        {
            k-=(size[ch[rt][0]]+1);
            rt=ch[rt][1];
        }
    }
    Splay(rt,goal);
}
inline int getMax(int rt)
{
    while(ch[rt][1])
        rt=ch[rt][1];
    return rt;
}
void DelRoot()
{
    int rt=root;
    if(ch[rt][0]==0)
    {
        pre[ch[rt][1]]=0;
        root=ch[rt][1];
        return ;
    }
    int rm=getMax(ch[rt][0]);
    Splay(rm,rt);
    ch[rm][1]=ch[rt][1];
    pre[ch[rt][1]]=rm;
    pre[rm]=0;
    root=rm;
    PushUp(root);
}
void Insert(int x,int y)
{
    RotateTo(x,0);
    int rt=root;
    int nt;
    if(ch[rt][0]==0)
    {
        Newnode(nt,rt,y);
        ch[rt][0]=nt;
        PushUp(rt);
        return;
    }
    int rm=getMax(ch[rt][0]);
    Splay(rm,rt);
    Newnode(nt,rm,y);
    ch[rm][1]=nt;
    PushUp(rm);
    PushUp(rt);
}
void build(int l,int r,int &rt,int fa)
{
    if(l>r) return ;
    int m=(l+r)>>1;
    Newnode(rt,fa,num[m]);
    build(l,m-1,ch[rt][0],rt);
    build(m+1,r,ch[rt][1],rt);
    PushUp(rt);
}
void Init()
{
    ch[0][0]=ch[0][1]=0;
    pre[0]=size[0]=0;
    pmax[0]=fmax[0]=smax[0]=-inf;
    val[0]=sum[0]=0;
    tot=root=0;
    Newnode(root,0,0);
    Newnode(ch[root][1],root,0);
    build(1,n,ch[ch[root][1]][0],ch[root][1]);
    PushUp(ch[root][1]);
    PushUp(root);
}
}splay_tree;
char rchar()
{
    char res;
    while(res=getchar(),!isalpha(res));
    return res;
}
void reads(int & x)
{
    char c;
    bool neg=false;
    while(((c=getchar())<'0'||c>'9')&&c!='-');
    if(c=='-')
    {
        neg=true;
        while((c=getchar())<'0'||c>'9');
    }
    x=c-'0';
    while(c=getchar(),c>='0'&&c<='9') x=x*10+c-'0';
    if(neg) x=-x;
}
void outs(int num)
{
     if(num<0)
     {
         putchar('-');
         num=-num;
     }
     int ans[10],top=0;
     while(num!=0)
     {
         ans[top++]=num%10;
         num/=10;
     }
     if(top==0)
     putchar('0');
     for(int i=top-1;i>=0;i--){
         char ch=ans[i]+'0';
         putchar(ch);
     }
     putchar('\n');
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    char c;
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;++i)
            reads(num[i]);
        splay_tree.Init();
        reads(q);
        int x,y;
        while(q--)
        {
            c=rchar();
            if(c=='I')
            {
                reads(x);reads(y);
                splay_tree.Insert(x,y);
            }
            else if(c=='D')
            {
                reads(x);
                splay_tree.RotateTo(x,0);
                splay_tree.DelRoot();
            }
            else if(c=='R')
            {
                reads(x);reads(y);
                splay_tree.RotateTo(x,0);
                splay_tree.val[splay_tree.root]=y;
                splay_tree.PushUp(splay_tree.root);
            }
            else
            {
                reads(x);reads(y);
                splay_tree.RotateTo(x-1,0);
                splay_tree.RotateTo(y+1,splay_tree.root);
                outs(splay_tree.smax[splay_tree.ch[splay_tree.ch[splay_tree.root][1]][0]]);
            }
        }
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值