bzoj 4034: [HAOI2015]树上操作(线段树+dfs序)
来源:互联网 发布:淘宝美工图片后期处理 编辑:程序博客网 时间:2024/05/22 17:42
操作 1 :把某个节点 x 的点权增加 a 。
操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 :询问某个节点 x 到根的路径中所有点的点权和。
用dfs序进栈+1出栈-1+线段树区间修改,查询维护
PS:线段树又写挂了,调试了很久啊~~
#include<iostream>#include<cstdio>#include<cmath>#include<cstring>#include<algorithm>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fod(i,a,b) for(int i=a;i>=b;i--)using namespace std;const int N=2e5+10,M=N*4;typedef long long ll;ll ls[M],rs[M],flag[M],sum[M],tag[M],a[N],n,m,ins[N],out[N],dfssum[N],cnt=0,mk[N];struct Edge{ int to,next; Edge(int to=0,int next=0):to(to),next(next){}}e[N<<1];int head[N],tot=0;void add_edge(int u,int v){e[++tot]=Edge(v,head[u]);head[u]=tot;}void dfs(int u,int fa){ ins[u]=++cnt;dfssum[cnt]=a[u];mk[cnt]=1; for(int i=head[u];i;i=e[i].next){ int id=e[i].to; if(id==fa)continue; dfs(id,u); } out[u]=++cnt;dfssum[cnt]=-a[u];mk[cnt]=-1;}void build(int k,int l,int r){ ls[k]=l;rs[k]=r;sum[k]=tag[k]=flag[k]=0; if(l==r) {sum[k]=dfssum[l];flag[k]=mk[l];return;} int mid=(l+r)>>1; build(k<<1,l,mid); build(k<<1|1,mid+1,r); sum[k]=sum[k<<1]+sum[k<<1|1]; flag[k]=flag[k<<1]+flag[k<<1|1];}void pushdown(int k){ tag[k<<1]+=tag[k]; tag[k<<1|1]+=tag[k]; sum[k<<1]+=(tag[k]*flag[k<<1]); sum[k<<1|1]+=(tag[k]*flag[k<<1|1]); tag[k]=0;}void update(int k,int ql,int qr,int delta){ if(ql==ls[k]&&qr==rs[k]){tag[k]+=delta;sum[k]+=(delta*flag[k]);return;} if(tag[k])pushdown(k); int mid=(ls[k]+rs[k])>>1; if(qr<=mid)update(k<<1,ql,qr,delta); else if(ql>mid)update(k<<1|1,ql,qr,delta); else{ update(k<<1,ql,mid,delta); update(k<<1|1,mid+1,qr,delta); } sum[k]=sum[k<<1]+sum[k<<1|1];}ll query(int k,int ql,int qr){ if(ql==ls[k]&&qr==rs[k]){return sum[k];} if(tag[k])pushdown(k); int mid=(ls[k]+rs[k])>>1; if(qr<=mid)return query(k<<1,ql,qr); else if(ql>mid)return query(k<<1|1,ql,qr); else return (query(k<<1,ql,mid)+query(k<<1|1,mid+1,qr));}int main(){ scanf("%lld%lld",&n,&m); fo(i,1,n) scanf("%lld",&a[i]); for(int u,v,i=1;i<n;i++) { scanf("%d%d",&u,&v); add_edge(u,v);add_edge(v,u); }dfs(1,0); build(1,1,cnt); for(int op,x,val,i=1;i<=m;i++){ scanf("%d",&op); if(op==1){scanf("%d%d",&x,&val);update(1,ins[x],ins[x],val);update(1,out[x],out[x],val);} if(op==2){scanf("%d%d",&x,&val);update(1,ins[x],out[x],val);} if(op==3){scanf("%d",&x);printf("%lld\n",query(1,1,ins[x]));} } return 0;}
阅读全文
0 0
- bzoj 4034: [HAOI2015]树上操作(线段树+dfs序)
- BZOJ 4034: [HAOI2015]树上操作【树链剖分】【DFS序】
- 【树链剖分(DFS序)+线段树】BZOJ4034(HAOI2015)[树上操作]题解
- BZOJ4034(HAOI2015)[树上操作]--线段树+DFS序
- |BZOJ 4034|树链剖分|线段树|[HAOI2015]树上操作
- BZOJ.4034 [HAOI2015]树上操作 ( 点权树链剖分 线段树 )
- BZOJ[4034][HAOI2015]树上操作 树链剖分+线段树
- BZOJ 4034 [HAOI2015]树上操作
- bzoj 4034 [HAOI2015]树上操作
- 【BZOJ 4034】 [HAOI2015]树上操作
- BZOJ 4034: [HAOI2015]树上操作
- bzoj 4034 [HAOI2015]树上操作
- BZOJ 4034 [HAOI2015]树上操作
- BZOJ 4034 HAOI2015 T2 DFS序+线段树
- bzoj 4034: [HAOI2015]树上操作(树链剖分+线段树区间更新)
- bzoj 4034: [HAOI2015]树上操作 (树链剖分)
- 【BZOJ】4034 [HAOI2015]树上操作 树链剖分
- BZOJ 4034 [HAOI2015]树上操作 树链剖分
- LeetCode 1. Two Sum
- 外部式css样式,写在单独的一个文件中
- ubuntu常用软件包deb的安装与卸载
- 返回值
- ISCSI配置方式
- bzoj 4034: [HAOI2015]树上操作(线段树+dfs序)
- 多态性
- Silver Cow Party
- android引导用户开启自启动权限
- 循环
- HDU --- 6181 Two Paths 2017第十场多校签到题【裸次短路】
- Git(Mac)
- 洛谷 1101 单词方阵
- 共享内存相关函数