ACM-ICPC 2018 焦作赛区网络预赛 E. Jiu Yuan Wants to Eat

---

分析

除了树剖没想到其他解法。
用线段树维护区间和，同时针对修改区间修改操作建立两个lazy标记，一个是\(lazy_{mul}\),另一个是\(lazy_{add}\)，代表区间里的数都需要先乘以\(lazy_{mul}\),再加上\(lazy_{add}\)。如果一个区间需要被重复标记，那么我们可以先把新的lazy标记施加在原有的lazy上。
如果是区间乘以val，那么就可以$ lazy_{mul}=lazy_{mul} \times val,lazy_{add}=lazy_{add}\times val $
对于区间取反操作可以转化成区间乘以-1，然后区间加上\(2^{64}-1\)

代码

#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ll;
const ll Mask=-1;
const int maxn=100050;
vector<int> G[maxn];
int sz[maxn],son[maxn],top[maxn],fa[maxn];
int dep[maxn];
void dfs(int x,int par) {
	sz[x]=1;
	fa[x]=par;
	for(auto y:G[x]) {
		if(y!=par) {
			dfs(y,x),sz[x]+=sz[y];
			if(son[x]==0||sz[y]>sz[son[x]]) son[x]=y;
		}
	}
}
int L[maxn],R[maxn],idx[maxn],tot;
void dfs2(int x,int par,int Top) {
	dep[x]=dep[par]+1;
	L[x]=++tot;
	idx[tot]=x;
	top[x]=Top;
	if(son[x]) dfs2(son[x],x,Top);
	for(auto y:G[x]) {
		if(y!=par && y!=son[x]) {
			dfs2(y,x,y);
		}
	}
	R[x]=tot;
}
ll sum[maxn*4],lazy_mul[maxn*4],lazy_add[maxn*4],wid[maxn*4];
void build(int l,int r,int id) {
	lazy_mul[id]=1,lazy_add[id]=0;
	wid[id]=r-l+1;
	sum[id]=0;
	if(l==r) return;
	int mid=(l+r)>>1;
	build(l,mid,id<<1);
	build(mid+1,r,id<<1|1);
}
void pushDown(int id) {
	if(wid[id]==1) {
		lazy_mul[id]=1,lazy_add[id]=0;
		return;
	}
	if(lazy_mul[id]==1&&lazy_add[id]==0) return;
	int x=(id<<1);
	lazy_mul[x]*=lazy_mul[id];
	lazy_add[x]*=lazy_mul[id];
	lazy_add[x]+=lazy_add[id];
	sum[x]=sum[x]*lazy_mul[id]+(lazy_add[id]*wid[x]);

	x=(id<<1|1);
	lazy_mul[x]*=lazy_mul[id];
	lazy_add[x]*=lazy_mul[id];
	lazy_add[x]+=lazy_add[id];
	sum[x]=sum[x]*lazy_mul[id]+(lazy_add[id]*wid[x]);
	lazy_mul[id]=1;
	lazy_add[id]=0;
}
void pushUp(int id) {
	sum[id]=sum[id<<1]+sum[id<<1|1];
}
void mul_update(int x,int y,ll val,int l,int r,int id) {
	pushDown(id);
	if(x<=l && y>=r) {
		lazy_mul[id]*=val;
		lazy_add[id]*=val;
		sum[id]*=val;
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid) mul_update(x,y,val,l,mid,id<<1);
	if(y>mid) mul_update(x,y,val,mid+1,r,id<<1|1);
	pushUp(id);
}
void add_update(int x,int y,ll val,int l,int r,int id) {
	pushDown(id);
	if(x<=l && y>=r) {
		lazy_add[id]+=val;
		sum[id]+=val*wid[id];
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid) add_update(x,y,val,l,mid,id<<1);
	if(y>mid) add_update(x,y,val,mid+1,r,id<<1|1);
	pushUp(id);
}
ll query(int x,int y,int l,int r,int id) {
	// cout<<l<<" "<<r<<" "<<sum[id]<<endl;
	pushDown(id);
	if(x<=l && y>=r) {
		//cout<<"in!\n";
		return sum[id];
	}
	int mid=(l+r)>>1;
	ll ans=0;
	if(x<=mid) /*(cout<<"go left\n",*/ ans+=query(x,y,l,mid,id<<1);
	if(y>mid) /*cout<<"go right\n",*/ ans+=query(x,y,mid+1,r,id<<1|1);
	return ans;
}
int main() {
	int n,q;
	while(scanf("%d", &n)!=EOF) {
		build(1,n,1);
		for(int i = 1; i <= n; ++i) G[i].clear(),sz[i]=top[i]=son[i]=0;
		for(int i = 2; i <= n; ++i) {
			int x;
			scanf("%d", &x);
			G[x].push_back(i);
		}
		scanf("%d", &q);
		tot=0;
		dfs(1,1);
		dfs2(1,1,1);
		while(q--) {
			int t,x,y;
			scanf("%d%d%d", &t,&x,&y);
			ll val;
			if(t==1) {
				scanf("%llu", &val);
				while(top[x]!=top[y]) {
					if(dep[top[x]]<dep[top[y]]) swap(x,y);
					mul_update(L[top[x]],L[x],val,1,n,1);
					x=fa[top[x]];
				}
				if(L[x]>L[y]) swap(x,y);
				mul_update(L[x],L[y],val,1,n,1);
			}
			else if(t==2) {
				scanf("%llu", &val);
				while(top[x]!=top[y]) {
					if(dep[top[x]]<dep[top[y]]) swap(x,y);
					add_update(L[top[x]],L[x],val,1,n,1);
					x=fa[top[x]];
				}
				if(L[x]>L[y]) swap(x,y);
				add_update(L[x],L[y],val,1,n,1);
			}
			else if(t==3) {
				while(top[x]!=top[y]) {
					if(dep[top[x]]<dep[top[y]]) swap(x,y);
					mul_update(L[top[x]],L[x],-1,1,n,1);
					add_update(L[top[x]],L[x],-1,1,n,1);
					x=fa[top[x]];
				}
				if(L[x]>L[y]) swap(x,y);
				mul_update(L[x],L[y],-1,1,n,1);
				add_update(L[x],L[y],-1,1,n,1);
			}
			else {
				ll ans=0;
				while(top[x]!=top[y]) {
					if(dep[top[x]]<dep[top[y]]) swap(x,y);
					ans+=query(L[top[x]],L[x],1,n,1);
					x=fa[top[x]];
				}
				if(L[x]>L[y]) swap(x,y);
				ans+=query(L[x],L[y],1,n,1);
				printf("%llu\n", ans);
			}
		}
	}
	return 0;
}