Description
背景:烁烁很喜欢爬树,这吓坏了树上的皮皮鼠。
题意:
给定一颗n个节点的树,边权均为1,初始树上没有皮皮鼠。
烁烁他每次会跳到一个节点u,把周围与他距离不超过d的节点各吸引出w只皮皮鼠。皮皮鼠会被烁烁吸引,所以会一直待在节点上不动。
烁烁很好奇,在当前时刻,节点u有多少个他的好朋友---皮皮鼠。
大意:
给一颗n个节点的树,边权均为1,初始点权均为0,m次操作:
Q x:询问x的点权。
M x d w:将树上与节点x距离不超过d的节点的点权均加上w。
Input
第一行两个正整数:n,m
接下来的n-1行,每行三个正整数u,v,代表u,v之间有一条边。
接下来的m行,每行给出上述两种操作中的一种。
Output
对于每个Q操作,输出当前x节点的皮皮鼠数量。
Sample Input
7 6
1 2
1 4
1 5
2 3
2 7
5 6
M 1 1 2
Q 5
M 2 2 3
Q 3
M 1 2 1
Q 2
1 2
1 4
1 5
2 3
2 7
5 6
M 1 1 2
Q 5
M 2 2 3
Q 3
M 1 2 1
Q 2
Sample Output
2
3
6
3
6
和震波哪题很像,也是线段树+动态点分治
只不过将操作换了,一样的原理
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<cmath> 6 using namespace std; 7 #define MAX 120000 8 struct Node 9 { 10 int next,to; 11 }edge[MAX<<1]; 12 int num,head[MAX]; 13 int dep[MAX],fa[MAX],pos,size[MAX],minsize,root; 14 int FA[MAX],son[MAX],top[MAX],Size; 15 int lazy[MAX*150]; 16 int ch[MAX*150][2],RT[MAX<<1],n,m; 17 int ans,val[MAX]; 18 bool vis[MAX]; 19 inline int gi() 20 { 21 int x=0,flag=1; 22 char ch=getchar(); 23 while (ch<‘0‘||ch>‘9‘) 24 { 25 if (ch==‘-‘) flag=-1; 26 ch=getchar(); 27 } 28 while (ch>=‘0‘&&ch<=‘9‘) 29 { 30 x=(x<<3)+(x<<1)+ch-‘0‘; 31 ch=getchar(); 32 } 33 return x*flag; 34 } 35 inline void add(int u,int v) 36 { 37 num++; 38 edge[num].next=head[u]; 39 head[u]=num; 40 edge[num].to=v; 41 } 42 void dfs1(int x,int pa) 43 { 44 size[x]=1; 45 dep[x]=dep[pa]+1; 46 for (int i=head[x];i;i=edge[i].next) 47 { 48 int v=edge[i].to; 49 if (v==pa) continue; 50 fa[v]=x; 51 dfs1(v,x); 52 size[x]+=size[v]; 53 if (size[son[x]]<size[v]) 54 son[x]=v; 55 } 56 } 57 void dfs2(int x,int tp) 58 { 59 top[x]=tp; 60 if (son[x]) 61 dfs2(son[x],tp); 62 for (int i=head[x];i;i=edge[i].next) 63 { 64 int v=edge[i].to; 65 if (v==fa[x]||v==son[x]) continue; 66 dfs2(v,v); 67 } 68 } 69 int lca(int x,int y) 70 { 71 while (top[x]!=top[y]) 72 { 73 if (dep[top[x]]<dep[top[y]]) swap(x,y); 74 x=fa[top[x]]; 75 } 76 if (dep[x]<dep[y]) return x; 77 return y; 78 } 79 int dis(int x,int y) 80 { 81 return dep[x]+dep[y]-(dep[lca(x,y)]<<1); 82 } 83 void get_root(int x,int pa) 84 { 85 size[x]=1; 86 int ret=0; 87 for (int i=head[x];i;i=edge[i].next) 88 { 89 int v=edge[i].to; 90 if (v==pa||vis[v]) continue; 91 get_root(v,x); 92 size[x]+=size[v]; 93 if (size[v]>ret) ret=size[v]; 94 } 95 if (Size-size[x]>ret) ret=Size-size[x]; 96 if (ret<minsize) minsize=ret,root=x; 97 } 98 void solve(int x,int pa) 99 { 100 vis[x]=1; 101 FA[x]=pa; 102 for (int i=head[x];i;i=edge[i].next) 103 { 104 int v=edge[i].to; 105 if (vis[v]) continue; 106 minsize=n;Size=size[v]; 107 get_root(v,x); 108 solve(root,x); 109 } 110 } 111 void update(int &rt,int l,int r,int L,int R,int w) 112 { 113 if (!rt) rt=++pos; 114 if (l>=L&&r<=R) 115 {lazy[rt]+=w;return;} 116 int mid=(l+r)>>1; 117 if (L<=mid) update(ch[rt][0],l,mid,L,R,w); 118 if (R>mid) update(ch[rt][1],mid+1,r,L,R,w); 119 } 120 int query(int rt,int l,int r,int x) 121 { 122 if (!rt) return 0; 123 if (l==r) 124 return lazy[rt]; 125 int mid=(l+r)>>1; 126 if (x<=mid) return query(ch[rt][0],l,mid,x)+lazy[rt]; 127 else return query(ch[rt][1],mid+1,r,x)+lazy[rt]; 128 } 129 void change(int u,int k,int w) 130 { 131 update(RT[u],0,n,0,k,w); 132 for (int i=u;FA[i];i=FA[i]) 133 { 134 int d=dis(u,FA[i]); 135 if (d>k) continue; 136 update(RT[FA[i]],0,n,0,k-d,w); 137 update(RT[i+n],0,n,0,k-d,w); 138 } 139 } 140 void ask(int u) 141 { 142 ans+=query(RT[u],0,n,0); 143 for (int i=u;FA[i];i=FA[i]) 144 { 145 int d=dis(u,FA[i]); 146 ans+=query(RT[FA[i]],0,n,d); 147 ans-=query(RT[i+n],0,n,d); 148 } 149 } 150 int main() 151 {int u,v; 152 char s[10]; 153 cin>>n>>m; 154 for (int i=1;i<=n-1;i++) 155 { 156 u=gi();v=gi(); 157 add(u,v);add(v,u); 158 } 159 dfs1(1,0); 160 dfs2(1,1); 161 minsize=Size=n; 162 get_root(1,0); 163 solve(root,0); 164 ans=0; 165 for (int i=1;i<=m;i++) 166 { 167 scanf("%s",s); 168 if (s[0]==‘Q‘) 169 { 170 ans=0; 171 int x=gi(); 172 ask(x); 173 printf("%d\n",ans); 174 } 175 else 176 { 177 int u=gi(),k=gi(),w=gi(); 178 change(u,k,w); 179 } 180 } 181 return 0; 182 } 183
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<cmath> using namespace std; #define MAX 120000 struct Node { int next,to; }edge[MAX<<1]; int num,head[MAX]; int dep[MAX],fa[MAX],pos,size[MAX],minsize,root; int FA[MAX],son[MAX],top[MAX],Size; int lazy[MAX*150]; int ch[MAX*150][2],RT[MAX<<1],n,m; int ans,val[MAX]; bool vis[MAX]; inline int gi() { int x=0,flag=1; char ch= getchar (); while (ch< ‘0‘ ||ch> ‘9‘ ) { if (ch== ‘-‘ ) flag=-1; ch= getchar (); } while (ch>= ‘0‘ &&ch<= ‘9‘ ) { x=(x<<3)+(x<<1)+ch- ‘0‘ ; ch= getchar (); } return x*flag; } inline void add( int u, int v) { num++; edge[num].next=head[u]; head[u]=num; edge[num].to=v; } void dfs1( int x, int pa) { size[x]=1; dep[x]=dep[pa]+1; for ( int i=head[x];i;i=edge[i].next) { int v=edge[i].to; if (v==pa) continue ; fa[v]=x; dfs1(v,x); size[x]+=size[v]; if (size[son[x]]<size[v]) son[x]=v; } } void dfs2( int x, int tp) { top[x]=tp; if (son[x]) dfs2(son[x],tp); for ( int i=head[x];i;i=edge[i].next) { int v=edge[i].to; if (v==fa[x]||v==son[x]) continue ; dfs2(v,v); } } int lca( int x, int y) { while (top[x]!=top[y]) { if (dep[top[x]]<dep[top[y]]) swap(x,y); x=fa[top[x]]; } if (dep[x]<dep[y]) return x; return y; } int dis( int x, int y) { return dep[x]+dep[y]-(dep[lca(x,y)]<<1); } void get_root( int x, int pa) { size[x]=1; int ret=0; for ( int i=head[x];i;i=edge[i].next) { int v=edge[i].to; if (v==pa||vis[v]) continue ; get_root(v,x); size[x]+=size[v]; if (size[v]>ret) ret=size[v]; } if (Size-size[x]>ret) ret=Size-size[x]; if (ret<minsize) minsize=ret,root=x; } void solve( int x, int pa) { vis[x]=1; FA[x]=pa; for ( int i=head[x];i;i=edge[i].next) { int v=edge[i].to; if (vis[v]) continue ; minsize=n;Size=size[v]; get_root(v,x); solve(root,x); } } void update( int &rt, int l, int r, int L, int R, int w) { if (!rt) rt=++pos; if (l>=L&&r<=R) {lazy[rt]+=w; return ;} int mid=(l+r)>>1; if (L<=mid) update(ch[rt][0],l,mid,L,R,w); if (R>mid) update(ch[rt][1],mid+1,r,L,R,w); } int query( int rt, int l, int r, int x) { if (!rt) return 0; if (l==r) return lazy[rt]; int mid=(l+r)>>1; if (x<=mid) return query(ch[rt][0],l,mid,x)+lazy[rt]; else return query(ch[rt][1],mid+1,r,x)+lazy[rt]; } void change( int u, int k, int w) { update(RT[u],0,n,0,k,w); for ( int i=u;FA[i];i=FA[i]) { int d=dis(u,FA[i]); if (d>k) continue ; update(RT[FA[i]],0,n,0,k-d,w); update(RT[i+n],0,n,0,k-d,w); } } void ask( int u) { ans+=query(RT[u],0,n,0); for ( int i=u;FA[i];i=FA[i]) { int d=dis(u,FA[i]); ans+=query(RT[FA[i]],0,n,d); ans-=query(RT[i+n],0,n,d); } } int main() { int u,v; char s[10]; cin>>n>>m; for ( int i=1;i<=n-1;i++) { u=gi();v=gi(); add(u,v);add(v,u); } dfs1(1,0); dfs2(1,1); minsize=Size=n; get_root(1,0); solve(root,0); ans=0; for ( int i=1;i<=m;i++) { scanf ( "%s" ,s); if (s[0]== ‘Q‘ ) { ans=0; int x=gi(); ask(x); printf ( "%d\n" ,ans); } else { int u=gi(),k=gi(),w=gi(); change(u,k,w); } } return 0; } |