【树链剖分(DFS序)+线段树】BZOJ4034(HAOI2015)[树上操作]题解

题目概述

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个操作，分为三种：

操作 1 ：把某个节点 x 的点权增加 a 。

操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。

操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

解题报告

操作1,3直接用树链剖分就可以解决，但是操作2什么鬼？

实际上很简单，因为树链剖分也在构造一组DFS序（只不过优先重儿子），所以用这组DFS序区间修改就行了。

示例程序

盗了以前的代码，发现巨丑……

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=100000,maxm=100000,MAXINT=((1<<30)-1)*2+1;

int n,te,E,num[maxn+5];
int lnk[maxn+5],son[2*maxm+5],nxt[2*maxm+5];
int si[maxn+5],dep[maxn+5],fa[maxn+5],sonH[maxn+5];
int top[maxn+5],Lt[maxn+5],Rt[maxn+5],who[maxn+5],tot;

bool Eoln(char ch) {return ch==10||ch==13||ch==EOF;}
int readi(int &x)
{
    int tot=0,f=1;char ch=getchar(),lst=ch;
    while ('9'<ch||ch<'0') {if (ch==EOF) return EOF;lst=ch;ch=getchar();}
    if (lst=='-') f=-f;
    while ('0'<=ch&&ch<='9') tot=tot*10+ch-48,ch=getchar();
    x=tot*f;
    return Eoln(ch);
}
void Add(int x,int y) {son[++E]=y;nxt[E]=lnk[x];lnk[x]=E;}
void Dfs(int x)
{
    dep[x]=dep[fa[x]]+1;si[x]=1;
    for (int j=lnk[x];j;j=nxt[j])
        if (son[j]!=fa[x])
        {
            fa[son[j]]=x;
            Dfs(son[j]);
            if (si[son[j]]>si[sonH[x]]) sonH[x]=son[j];
            si[x]+=si[son[j]];
        }
}
void HLD(int x,int lst)
{
    top[x]=lst;Lt[x]=++tot;who[tot]=x;
    if (sonH[x]) HLD(sonH[x],lst);
    for (int j=lnk[x];j;j=nxt[j])
        if (son[j]!=fa[x]&&son[j]!=sonH[x]) HLD(son[j],son[j]);
    Rt[x]=tot;
}
struct SegmentTree
{
    int l[4*maxn+5],r[4*maxn+5];LL sum[4*maxn+5],tem[4*maxn+5];
    void Build(int id,int L,int R)
    {
        l[id]=L;r[id]=R;
        if (L==R) {sum[id]=num[who[L]];return;}
        int mid=L+(R-L>>1);
        Build(id*2,L,mid);Build(id*2+1,mid+1,R);
        Pushup(id);
    }
    void Pushup(int id) {sum[id]=sum[id*2]+sum[id*2+1];}
    void Pushdown(int id)
    {
        LL now=tem[id];tem[id]=0;
        sum[id*2]+=now*(r[id*2]-l[id*2]+1);tem[id*2]+=now;
        sum[id*2+1]+=now*(r[id*2+1]-l[id*2+1]+1);tem[id*2+1]+=now;
    }
    void Update(int id,int L,int R,int x)
    {
        if (R<l[id]||r[id]<L) return;
        if (L<=l[id]&&r[id]<=R) {sum[id]+=(LL)x*(r[id]-l[id]+1);tem[id]+=x;return;}
        Pushdown(id);
        Update(id*2,L,R,x);Update(id*2+1,L,R,x);
        Pushup(id);
    }
    LL Ask_sum(int id,int L,int R)
    {
        if (R<l[id]||r[id]<L) return 0;
        if (L<=l[id]&&r[id]<=R) return sum[id];
        Pushdown(id);
        return Ask_sum(id*2,L,R)+Ask_sum(id*2+1,L,R);
    }
};
SegmentTree tr;
LL Ask_sum(int x,int y)
{
    LL sum=0;
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        sum+=tr.Ask_sum(1,Lt[top[x]],Lt[x]);
        x=fa[top[x]];
    }
    if (Lt[x]>Lt[y]) swap(x,y);
    sum+=tr.Ask_sum(1,Lt[x],Lt[y]);
    return sum;
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    readi(n);readi(te);
    for (int i=1;i<=n;i++) readi(num[i]);
    for (int i=1;i<=n-1;i++)
    {
        int x,y;readi(x);readi(y);
        Add(x,y);Add(y,x);
    }
    Dfs(1);HLD(1,1);tr.Build(1,1,n);
    while (te--)
    {
        int td,x,y;readi(td);readi(x);
        if (td==1) readi(y),tr.Update(1,Lt[x],Lt[x],y); else
        if (td==2) readi(y),tr.Update(1,Lt[x],Rt[x],y); else
        printf("%lld\n",Ask_sum(1,x));
    }
    return 0;
}