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棵线段树进行区间修改。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
#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); } } } } }