BZOJ 4034 [HAOI2015]树上操作(欧拉序+线段树)
题意：
有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个
操作，分为三种：
操作 1 ：把某个节点 x 的点权增加 a 。
操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。
思路：
处理出这棵树的欧拉序，入栈时为这个点的正权，出栈时为这个点的负权
对于操作1，对x入栈点加a，出栈点减a
对于操作2，对x入栈点到x出栈点所有的点执行操作1
对于操作3，即查询点1的入栈点到x入栈点的点权和
在正常的区间加线段树中，有一个add，add对区间和sum[root]的贡献为(r-l+1)*add
对这一题，我们记入栈点的flg为1，出栈点的为-1，那么在flg求和的情况下，add对区间和的贡献为flg[root]*add
正常搞线段树即可
代码：
#include<iostream> #include<cstdio> #include<algorithm> #include<cmath> #include<cstring> #include<string> #include<stack> #include<queue> #include<deque> #include<set> #include<vector> #include<map> #include<functional> #define fst first #define sc second #define pb push_back #define mem(a,b) memset(a,b,sizeof(a)) #define lson l,mid,root<<1 #define rson mid+1,r,root<<1|1 #define lc root<<1 #define rc root<<1|1 #define lowbit(x) ((x)&(-x)) using namespace std; typedef double db; typedef long double ldb; typedef long long ll; typedef long long LL; typedef unsigned long long ull; typedef pair<int,int> PI; typedef pair<ll,ll> PLL; const db eps = 1e-6; const int mod = 998244353; const int maxn = 2e6+100; const int maxm = 2e6+100; const int inf = 0x3f3f3f3f; const db pi = acos(-1.0); int n, m; ll a[maxn]; vector<int>v[maxn]; int tot; int in[maxn],out[maxn];//树上i的出、入在rk位置 ll rk[maxn]; int vis[maxn];//in 1 , out 0 void dfs(int x, int fa){ ++tot;in[x]=tot;rk[tot]=a[x];vis[tot]=1; for(int i = 0; i < (int)v[x].size(); i++){ int y = v[x][i]; if(y!=fa)dfs(y,x); } ++tot;out[x]=tot;rk[tot]=-a[x];vis[tot]=0; } ll flg[maxn],lazy[maxn]; ll sum[maxn]; void build(int l, int r, int root){ if(l==r){ if(vis[l])flg[root]=1; else flg[root]=-1; sum[root]=rk[l]; return; } int mid = (l+r)>>1; build(lson); build(rson); sum[root]=sum[lc]+sum[rc]; flg[root]=flg[lc]+flg[rc]; } void pushdown(int l, int r, int root){ if(!lazy[root])return; lazy[lc]+=lazy[root]; lazy[rc]+=lazy[root]; sum[lc]+=lazy[root]*flg[lc]; sum[rc]+=lazy[root]*flg[rc]; lazy[root]=0; return; } void update(int x, int y, int val, int l, int r, int root){ int mid = (l+r)>>1; if(x<=l&&r<=y){ lazy[root]+=val; sum[root]+=val*flg[root]; return; } pushdown(l, r, root); if(x<=mid)update(x,y,val,lson); if(y>mid)update(x,y,val,rson); sum[root]=sum[lc]+sum[rc]; return; } ll query(int x, int y, int l, int r, int root){ int mid = (l+r)>>1; if(x<=l&&r<=y)return sum[root]; pushdown(l, r, root); ll ans = 0; if(x<=mid)ans+=query(x,y,lson); if(y>mid)ans+=query(x,y,rson); return ans; } int main() { scanf("%d %d", &n, &m); for(int i = 1; i <= n; i++){ scanf("%lld", &a[i]); } for(int i = 1; i <= n-1; i++){ int x, y; scanf("%d %d" ,&x, &y); v[x].pb(y); v[y].pb(x); } dfs(1,-1); build(1,tot,1); while(m--){ int op, x, y; scanf("%d", &op); if(op==1){ scanf("%d %d" ,&x ,&y); update(in[x],in[x],y,1,tot,1); update(out[x],out[x],y,1,tot,1); } else if(op==2){ scanf("%d %d", &x ,&y); update(in[x],out[x],y,1,tot,1); } else{ scanf("%d", &x); printf("%lld\n",query(1,in[x],1,tot,1)); } } return 0; } /* 5 5 1 2 3 4 5 1 2 1 4 2 3 2 5 3 3 1 2 1 3 5 2 1 2 3 3 */
