首页 > 代码库 > Hdu 3699 Aragorn's Story (树链剖分)
Hdu 3699 Aragorn's Story (树链剖分)
题目大意:
对一颗树上进行路径加减,然后询问单点的值。
思路分析:
简单的树链剖分模板题。
#include <cstdio> #include <iostream> #include <cstring> #include <algorithm> #pragma comment(linker,"/STACk:10240000,10240000") #define maxn 50005 #define lson num<<1,s,mid #define rson num<<1|1,mid+1,e using namespace std; int next[maxn<<1],to[maxn<<1],head[maxn],tot;//临界表 int pre[maxn],root[maxn],siz[maxn],son[maxn],w[maxn],dep[maxn],id;//原树的父亲 链上的根 siz 有最大siz的子树 重新分配的id 深度 getid中来计数的 int sum[maxn<<2],add[maxn<<2];//线段树变量 int n; void init() { pre[1]=0; dep[1]=0; tot=0;id=0; memset(head,0,sizeof head); memset(sum,0,sizeof sum); memset(add,0,sizeof add); } void addedge(int u,int v) { tot++; next[tot]=head[u]; to[tot]=v; head[u]=tot; } void dfs(int now) { son[now]=0; siz[now]=1; for(int p =head[now]; p ; p=next[p]) { int t=to[p]; if(t!=pre[now]) { pre[t]=now; dep[t]=dep[now]+1; dfs(t); if(siz[t]>siz[son[now]])son[now]=t; siz[now]+=siz[t]; } } } void getid(int now,int rt) { w[now]=++id; root[now]=rt; if(son[now])getid(son[now],rt); for(int p = head[now] ; p ; p=next[p]) { int t=to[p]; if(t!=son[now]&&t!=pre[now]) getid(t,t); } } void pushdown(int num,int s,int e) { if(add[num]) { int mid=(s+e)>>1; sum[num<<1]+=(mid-s+1)*add[num]; sum[num<<1|1]+=(e-mid)*add[num]; add[num<<1]+=add[num]; add[num<<1|1]+=add[num]; add[num]=0; } } void update(int num,int s,int e,int l,int r,int val) { if(l<=s && r>=e) { sum[num]+=(e-s+1)*val; add[num]+=val; return; } pushdown(num,s,e); int mid=(s+e)>>1; if(l<=mid)update(lson,l,r,val); if(r>mid)update(rson,l,r,val); } int query(int num,int s,int e,int pos) { if(s==e)return sum[num]; pushdown(num,s,e); int mid=(s+e)>>1; if(pos<=mid)return query(lson,pos); else return query(rson,pos); } void work(int x,int y,int val) { while(root[x]!=root[y]) { if(dep[root[x]]<dep[root[y]])swap(x,y); update(1,1,n,w[root[x]],w[x],val); x=pre[root[x]]; } if(dep[x]>dep[y])swap(x,y); update(1,1,n,w[x],w[y],val); } int save[maxn]; int main() { int m,Q; while(scanf("%d%d%d",&n,&m,&Q)!=EOF) { init(); for(int i=1;i<=n;i++) scanf("%d",&save[i]); for(int i=1;i<=m;i++) { int s,e; scanf("%d%d",&s,&e); addedge(s,e); addedge(e,s); } dfs(1); getid(1,1); for(int i=1;i<=n;i++) update(1,1,n,w[i],w[i],save[i]); char str[5]; while(Q--) { scanf("%s",str); if(str[0]=='I') { int a,b,c; scanf("%d%d%d",&a,&b,&c); work(a,b,c); } else if(str[0]=='D') { int a,b,c; scanf("%d%d%d",&a,&b,&c); work(a,b,-c); } else { int a; scanf("%d",&a); printf("%d\n",query(1,1,n,w[a])); } } } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。