题目
分析
点分树+线段树
首先我们可以发现,这个和模板题的区别就是那里是单点修改区间查询,但是这里是区间修改单点查询。
于是在线段树上把这两个操作的对应区间反过来就好了。
具体可以见代码。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define M 100010
#define ls ch[node][0]
#define rs ch[node][1]
using namespace std;
int read() {
char ch=getchar();int x=0;
while(ch>'9'||ch<'0') ch=getchar();
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x;
}
int n,m,num,cnt,root,S;
int sz[M],size[M],top[M],fa[M],deep[M],son[M],head[M],maxn[M],f[M];
int rt1[M],rt2[M],val[M<<7],ch[M<<7][2];bool vis[M];
struct point{int next,to;}e[M<<1];
void add(int from,int to) {
e[++num].next=head[from];
e[num].to=to;
head[from]=num;
}
void dfs1(int x) {
sz[x]=1;deep[x]=deep[fa[x]]+1;
for(int i=head[x];i;i=e[i].next) {
int to=e[i].to;
if(fa[x]==to) continue;
fa[to]=x,dfs1(to),sz[x]+=sz[to];
if(sz[to]>sz[son[x]]) son[x]=to;
}
}
void dfs2(int x,int tp) {
top[x]=tp;
if(son[x]) dfs2(son[x],tp);
for(int i=head[x];i;i=e[i].next)
if(e[i].to!=son[x]&&e[i].to!=fa[x])
dfs2(e[i].to,e[i].to);
}
int lca(int x,int y) {
while(top[x]!=top[y]) {
if(deep[top[x]]<deep[top[y]]) swap(x,y);
x=fa[top[x]];
}
return deep[x]>deep[y]?y:x;
}
int dis(int x,int y) {
return deep[x]+deep[y]-2*deep[lca(x,y)];
}
void getroot(int x,int fa) {
size[x]=1;maxn[x]=0;
for(int i=head[x];i;i=e[i].next) {
int to=e[i].to;
if(to==fa||vis[to]) continue;
getroot(to,x);
size[x]+=size[to];
maxn[x]=max(maxn[x],size[to]);
}
maxn[x]=max(maxn[x],S-size[x]);
if(maxn[x]<maxn[root]) root=x;
}
void solve(int x,int fa) {
vis[x]=true;f[x]=fa;
for(int i=head[x];i;i=e[i].next) {
int to=e[i].to;
if(vis[to]) continue;
root=0,S=size[to],getroot(to,0);
solve(root,x);
}
}
void change(int &node,int l,int r,int l1,int r1,int v) {
if(!node) node=++cnt;
if(l1<=l&&r1>=r) {
val[node]+=v;return;
}
int mid=(l+r)/2;
if(l1<=mid) change(ls,l,mid,l1,r1,v);
if(r1>mid) change(rs,mid+1,r,l1,r1,v);
}
int query(int node,int l,int r,int x) {
if(!node) return 0;
if(l==r) return val[node];
int mid=(l+r)/2;
if(x<=mid) return val[node]+query(ls,l,mid,x);
else return val[node]+query(rs,mid+1,r,x);
}
void update(int x,int d,int v) {
change(rt1[x],0,n,0,d,v);
for(int i=x;f[i];i=f[i]) {
int dt=dis(x,f[i]);
if(dt>d) continue;
change(rt1[f[i]],0,n,0,d-dt,v);
change(rt2[i],0,n,0,d-dt,v);
}
}
int ask(int x) {
int res=query(rt1[x],0,n,0);
for(int i=x;f[i];i=f[i]) {
int dt=dis(x,f[i]);
res+=query(rt1[f[i]],0,n,dt)-query(rt2[i],0,n,dt);
}
return res;
}
int main() {
n=read();m=read();
for(int i=1;i<n;i++) {
int u=read(),v=read();
add(u,v),add(v,u);
}
dfs1(1),dfs2(1,1);
maxn[0]=S=n;
getroot(1,0),solve(root,0);
for(int i=1;i<=m;i++) {
char s[10];scanf("%s",s);
if(s[0]=='M') {
int x=read(),d=read(),w=read();
update(x,d,w);
}
else {
int x=read();
printf("%d
",ask(x));
}
}
return 0;
}