zoj 3765 Lights(伸展树)

题意:最开始有N个灯,有Q个操作。操作有一下几种类型:①查询区间[L,R]中灯的状态为status的所有灯的gcd②在第i位后插入一个值为val,状态为status的灯③删除第i个灯④第i个灯是开的那么关掉,否则打开⑤修改第i个灯的值。

思路:用伸展树暴力点写就行了。用数组记录当前区间状态为0和1的gcd的值。对于0结点,这些值赋值为-1就行了,在gcd中修改一下,对-1进行处理就行了。剩下的都是些基本操作。。。


代码:

#include <iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#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=300000+10;
int ch[maxn][2],pre[maxn],size[maxn],val[maxn],st[maxn];
int gcdst[maxn][2],root,tot;
int num[200010],fs[200010];
int gcd(int a,int b)
{
    if(a==-1) return b;
    if(b==-1) return a;
    int c;
    while(b)
    {
        c=a%b;
        a=b;
        b=c;
    }
    return a;
}
void NewNode(int &rt,int fa,int v,int state)
{
    rt=++tot;
    ch[rt][0]=ch[rt][1]=0;
    size[rt]=1;val[rt]=v;
    pre[rt]=fa;
    gcdst[rt][state]=v;gcdst[rt][state^1]=-1;
    st[rt]=state;
}
void PushUp(int rt)
{
    size[rt]=size[ch[rt][0]]+size[ch[rt][1]]+1;
    gcdst[rt][0]=gcd(gcdst[ch[rt][0]][0],gcdst[ch[rt][1]][0]);
    gcdst[rt][1]=gcd(gcdst[ch[rt][0]][1],gcdst[ch[rt][1]][1]);
    gcdst[rt][st[rt]]=gcd(gcdst[rt][st[rt]],val[rt]);
}
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);
}
void build(int l,int r,int &rt,int fa)
{
    if(l>r) return ;
    int m=(l+r)>>1;
    NewNode(rt,fa,num[m],fs[m]);
    build(l,m-1,ch[rt][0],rt);
    build(m+1,r,ch[rt][1],rt);
    PushUp(rt);
}
int Query(int L,int R,int state)
{
    RotateTo(L-1,0);
    RotateTo(R+1,root);
    return gcdst[ch[ch[root][1]][0]][state];
}
void Insert(int pos,int v,int state)
{
    RotateTo(pos,0);
    if(ch[root][0]==0)
    {
        NewNode(ch[root][0],root,v,state);
        PushUp(root);
        return ;
    }
    int rc=ch[root][0];
    while(ch[rc][1])
        rc=ch[rc][1];
    Splay(rc,root);
    NewNode(ch[rc][1],rc,v,state);
    PushUp(rc);
    PushUp(root);
}
void Delete(int pos)
{
    RotateTo(pos,0);
    if(ch[root][0]==0)
    {
        root=ch[root][1];
        pre[root]=0;
        return ;
    }
    int rc=ch[root][0];
    while(ch[rc][1]) rc=ch[rc][1];
    Splay(rc,root);
    int rt=ch[root][1];
    ch[rc][1]=rt;
    pre[rt]=rc;
    root=rc;
    pre[rc]=0;
    PushUp(root);
}
void Changes(int pos)
{
    RotateTo(pos,0);
    st[root]^=1;
    PushUp(root);
}
void Modify(int pos,int v)
{
    RotateTo(pos,0);
    val[root]=v;
    PushUp(root);
}
void Init(int n)
{
    for(int i=0;i<n;++i)
        scanf("%d%d",&num[i],&fs[i]);
    ch[0][0]=ch[0][1]=0;
    pre[0]=size[0]=0;val[0]=-1;
    gcdst[0][0]=gcdst[0][1]=-1;
    root=tot=0;
    NewNode(root,0,-1,0);
    NewNode(ch[root][1],root,-1,0);
    build(0,n-1,ch[ch[root][1]][0],ch[root][1]);
    PushUp(ch[root][1]);
    PushUp(root);
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int n,q;
    char str[5];
    while(~scanf("%d%d",&n,&q))
    {
        Init(n);
        while(q--)
        {
            scanf("%s",str);
            int L,R,x,s;
            if(str[0]=='Q')
            {
                scanf("%d%d%d",&L,&R,&s);
                printf("%d\n",Query(L,R,s));
            }
            else if(str[0]=='I')
            {
                scanf("%d%d%d",&L,&x,&s);
                Insert(L+1,x,s);
            }
            else if(str[0]=='D')
            {
                scanf("%d",&L);
                Delete(L);
            }
            else if(str[0]=='R')
            {
                scanf("%d",&L);
                Changes(L);
            }
            else
            {
                scanf("%d%d",&L,&x);
                Modify(L,x);
            }
        }
    }
    return 0;
}




你可能感兴趣的:(DataStruct)