FZU2176（二维线段树+dfs）

传送门：easy problem

题意：给定一棵n个节点以1为根的树，初始每个节点的值为0，现在我们要在树上进行一些操作，操作有两种类型。

1 x val 表示对以x为根的子树的每个点进行加权操作（我们定义每个节点的深度为每个节点到根1的距离），如果 y是以x为根的子树中的点那么 y节点的权值增加 (（dep[y]-dep[x]）%k+1)*val 其中dep[y]表示y节点的深度，k为一个常数(1<=k<=5)

2 x 查询当前x节点的权值。

分析：这题用树链剖分有点大材小用，直接一个dfs将每点遍历完又回到该点重新标号映射到线段树上，然后每修改该点及它的子节点时在线段树上操作。由题意可发现，每隔k个深度权值增加是一样的，而k又很小，因此用k棵线段树分别维护整段区间内深度模k的点余x（0<x<k）的点，然后修改区间时分别给区间内模k为0,1...k-1的点修改，也就是给k棵线段树进行区间修改。

#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <cstdlib>
#include <stack>
#include <vector>
#include <set>
#include <map>
#define LL long long
#define mod 100000000
#define inf 0x3f3f3f3f
#define eps 1e-6
#define N 50010
#define FILL(a,b) (memset(a,b,sizeof(a)))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define PII pair<int,int>
using namespace std;
struct edge
{
    int v,next;
    edge(){}
    edge(int v,int next):v(v),next(next){}
}e[N];
int head[N],vis[N],tot;
int col[5][N<<2],num;
int st[N],ed[N],dep[N];
int n,m,k;
void init()
{
    FILL(head,-1);
    FILL(vis,0);
    tot=0;
}
void addedge(int u,int v)
{
    e[tot]=edge(v,head[u]);
    head[u]=tot++;
}
void dfs(int u,int fa)
{
    st[u]=++num;
    for(int i=head[u];~i;i=e[i].next)
    {
        int v=e[i].v;
        dep[v]=dep[u]+1;
        dfs(v,u);
    }
    ed[u]=num;
}
void build(int l,int r,int rt)
{
    for(int i=0;i<k;i++)col[i][rt]=0;
    if(l==r)return;
    int m=(l+r)>>1;
    build(lson);
    build(rson);
}
void Pushdown(int rt)
{
    for(int i=0;i<k;i++)
    if(col[i][rt])
    {
        col[i][rt<<1]+=col[i][rt];
        col[i][rt<<1|1]+=col[i][rt];
        col[i][rt]=0;
    }
}
void update(int L,int R,int s,int c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        col[s][rt]+=c;
        return;
    }
    Pushdown(rt);
    int m=(l+r)>>1;
    if(L<=m)update(L,R,s,c,lson);
    if(m<R)update(L,R,s,c,rson);
}
int query(int pos,int s,int l,int r,int rt)
{
    if(l==r)
    {
        return col[s][rt];
    }
    Pushdown(rt);
    int m=(l+r)>>1;
    if(pos<=m)return query(pos,s,lson);
    else return query(pos,s,rson);
}
int main()
{

    int t,a,b,op,cas=1;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&k);
        init();
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            addedge(a,b);
        }
        num=0;dep[1]=0;
        dfs(1,-1);
        build(1,num,1);
        printf("Case#%d:\n",cas++);
        while(m--)
        {
            scanf("%d",&op);
            if(op==2)
            {
                scanf("%d",&a);
                printf("%d\n",query(st[a],(dep[a]+1)%k,1,num,1));
            }
            else
            {
                scanf("%d%d",&a,&b);
                for(int i=0;i<k;i++)
                {
                    int s=((dep[a]+1)%k+i)%k;
                    update(st[a],ed[a],s,b*(i+1),1,num,1);
                }
            }
        }
    }
}