BZOJ4034(HAOI2015)[树上操作]--线段树+DFS序

【链接】
bzoj4034

【解题报告】

此题好像是树剖模板，但是好像线段树+DFS序也可以解决。

只需要在DFS序中，对于一个点的进栈与出栈，只需要标记进栈为正，出栈为负，就可以解决了。

#include<cstdio>
#include<cstring>
#define LL long long
using namespace std;
const int maxn=100005,maxm=200005,maxv=800005;
int n,m,tot,cnt,a[maxn],lnk[maxn],son[maxm],nxt[maxm],in[maxm],out[maxm],p[maxm];
bool vis[maxm];
struct Tree
{
    LL s[maxv],tag[maxv],flg[maxv];
    void Pushup(int d) {s[d]=s[d<<1]+s[d<<1|1];}
    void Pushdown(int d)
    {
        if (!tag[d]) return;
        s[d<<1]+=(LL)tag[d]*flg[d<<1]; s[d<<1|1]+=(LL)tag[d]*flg[d<<1|1];
        tag[d<<1]+=tag[d]; tag[d<<1|1]+=tag[d]; tag[d]=0;
    }
    void Build(int d,int l,int r)
    {
        if (l==r) {s[d]=p[l]; tag[d]=0; if (vis[l]) flg[d]=1; else flg[d]=-1; return;}
        int mid=(r-l>>1)+l;
        Build(d<<1,l,mid); Build(d<<1|1,mid+1,r);
        Pushup(d); flg[d]=flg[d<<1]+flg[d<<1|1];
    }
    void Insert(int d,int l,int r,int x,int y,int p)
    {
        if (l==x&&r==y) {s[d]+=p*flg[d]; tag[d]+=p; return;}
        int mid=(r-l>>1)+l; Pushdown(d);
        if (y<=mid) Insert(d<<1,l,mid,x,y,p);
         else if (x>mid) Insert(d<<1|1,mid+1,r,x,y,p);
          else Insert(d<<1,l,mid,x,mid,p),Insert(d<<1|1,mid+1,r,mid+1,y,p);
        Pushup(d);
    }
    LL Query(int d,int l,int r,int x,int y)
    {
        if (l==x&&r==y) return s[d];
        int mid=(r-l>>1)+l; Pushdown(d);
        if (y<=mid) return Query(d<<1,l,mid,x,y);
         else if (x>mid) return Query(d<<1|1,mid+1,r,x,y);
          else return Query(d<<1,l,mid,x,mid)+Query(d<<1|1,mid+1,r,mid+1,y);
    }
}tr;
inline int Read()
{
    int res=0,f=1;
    char ch=getchar(),cc=ch;
    while (ch<'0'||ch>'9') cc=ch,ch=getchar();
    if (cc=='-') f=-1;
    while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=getchar();
    return res*f;
}
void Add(int x,int y) {son[++tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;}
void Dfs(int x,int fa)
{
    p[in[x]=++cnt]=a[x]; vis[cnt]=1;
    for (int j=lnk[x]; j; j=nxt[j])
     if (son[j]!=fa) Dfs(son[j],x);
    p[out[x]=++cnt]=-a[x]; vis[cnt]=0;
}
int main()
{
    freopen("4034.in","r",stdin);
    freopen("4034.out","w",stdout);
    n=Read(); m=Read(); tot=0; cnt=0;
    memset(lnk,0,sizeof(lnk));
    for (int i=1; i<=n; i++) a[i]=Read();
    for (int i=1,x,y; i<n; i++) x=Read(),y=Read(),Add(x,y),Add(y,x);
    Dfs(1,0); tr.Build(1,1,2*n);
    for (int i=1; i<=m; i++)
    {
        int z=Read(),x=Read(),p;
        if (z==1) p=Read(),tr.Insert(1,1,2*n,in[x],in[x],p),tr.Insert(1,1,2*n,out[x],out[x],p);
         else if (z==2) p=Read(),tr.Insert(1,1,2*n,in[x],out[x],p);
          else printf("%lld\n",tr.Query(1,1,2*n,1,in[x]));
    }
    return 0;
}