luogu_P3313 [SDOI2014]旅行

Posted 2021-01-28 papercloud

传送门

Solution

第二次学习可持久化线段树

打了一道裸题来练习一下……

• 对于每个宗教都可以开一个主席树
• 基础操作
• 树剖lca

Code?

#include<bits/stdc++.h>
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}
#define MN 100005
#define MM 8000005
int w[MN],c[MN],N;
struct edge{int to,nex;}e[MN<<1];
int hr[MN],en;
inline void ins(int f,int t)
{
    e[++en]=(edge){t,hr[f]};hr[f]=en;
    e[++en]=(edge){f,hr[t]};hr[t]=en;
}
int fa[MN],dep[MN],mx[MN],siz[MN],pos[MN],top[MN];
void dfs1(int x,int f)
{
    dep[x]=dep[fa[x]=f]+(siz[x]=1);register int i;
    for(i=hr[x];i;i=e[i].nex) if(f^e[i].to) dfs1(e[i].to,x),siz[e[i].to]>siz[mx[x]]?mx[x]=e[i].to:0,siz[x]+=siz[e[i].to];
}
int dind=0;
void dfs2(int x,int tp)
{
    pos[x]=++dind;top[x]=tp;register int i;
    if(mx[x]) dfs2(mx[x],tp);
    for(i=hr[x];i;i=e[i].nex) if((fa[x]^e[i].to)&&(mx[x]^e[i].to)) dfs2(e[i].to,e[i].to);
}
#define mid ((l+r)>>1)
int Max[MM],Sum[MM],ls[MM],rs[MM],rt[MN],sz;
int query_m(int x,int l,int r,int a,int b)
{
    if(!x) return 0;
    if(a==l&&r==b) return Max[x];
    if(b<=mid) return query_m(ls[x],l,mid,a,b);
    if(a>mid) return query_m(rs[x],mid+1,r,a,b);
    return max(query_m(ls[x],l,mid,a,mid),query_m(rs[x],mid+1,r,mid+1,b));
}
int query_s(int x,int l,int r,int a,int b)
{
    if(!x) return 0;
    if(a==l&&r==b) return Sum[x];
    if(b<=mid) return query_s(ls[x],l,mid,a,b);
    if(a>mid) return query_s(rs[x],mid+1,r,a,b);
    return query_s(ls[x],l,mid,a,mid)+query_s(rs[x],mid+1,r,mid+1,b);
}
inline void pushup(int x){Max[x]=max(Max[ls[x]],Max[rs[x]]);Sum[x]=Sum[ls[x]]+Sum[rs[x]];}
void Modify(int &x,int l,int r,int a,int v)
{
    if(!x) x=++sz;
    if(l==r) return (void)(Max[x]=Sum[x]=v);
    if(a<=mid) Modify(ls[x],l,mid,a,v);
    else Modify(rs[x],mid+1,r,a,v);
    pushup(x);
}
int QS(int root,int x,int y)
{
    int res=0;
    while(top[x]^top[y])
    {
        if(dep[top[x]]>dep[top[y]]) res+=query_s(root,1,N,pos[top[x]],pos[x]),x=fa[top[x]];
        else res+=query_s(root,1,N,pos[top[y]],pos[y]),y=fa[top[y]];
    }
    res+=query_s(root,1,N,min(pos[x],pos[y]),max(pos[x],pos[y]));
    return res;
}
int QM(int root,int x,int y)
{
    int res=0;
    while(top[x]^top[y])
    {
        if(dep[top[x]]>dep[top[y]]) res=max(res,query_m(root,1,N,pos[top[x]],pos[x])),x=fa[top[x]];
        else res=max(res,query_m(root,1,N,pos[top[y]],pos[y])),y=fa[top[y]];
    }
    res=max(res,query_m(root,1,N,min(pos[x],pos[y]),max(pos[x],pos[y])));
    return res;
}
int main()
{
    register int i,Q,x,y;register char opt[10];
    N=read(),Q=read();
    for(i=1;i<=N;++i) w[i]=read(),c[i]=read();
    for(i=1;i<N;++i) x=read(),ins(x,read());
    dfs1(1,0);dfs2(1,1);
    for(i=1;i<=N;++i) Modify(rt[c[i]],1,N,pos[i],w[i]);
    while(Q--)
    {
        scanf("%s",opt+1);x=read();y=read();
        if(opt[1]=='C'&&opt[2]=='C') Modify(rt[c[x]],1,N,pos[x],0),Modify(rt[c[x]=y],1,N,pos[x],w[x]);
        if(opt[1]=='C'&&opt[2]=='W') Modify(rt[c[x]],1,N,pos[x],w[x]=y);
        if(opt[1]=='Q'&&opt[2]=='S') printf("%d
",QS(rt[c[y]],x,y));
        if(opt[1]=='Q'&&opt[2]=='M') printf("%d
",QM(rt[c[y]],x,y));
    }
    return 0;
}

---

Blog来自PaperCloud，未经允许，请勿转载，TKS！