树剖+线段树+矩阵求调
查看原帖
树剖+线段树+矩阵求调
477032
diandian2020楼主2023/7/16 21:17
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=1e5+9,M=2e5+9,INF=1e9+1;
int n,Q,w[N];
int h[N],e[M],ne[M],idx;
int dep[N],fa[N],sz[N],son[N],top[N];
int dfn[N],nw[N],timestamp;
void add(int a,int b){
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs1(int u,int p){
	dep[u]=dep[p]+1,fa[u]=p,sz[u]=1;
	for(int i=h[u];~i;i=ne[i]){
		int v=e[i];
		if(v==p) continue;
		dfs1(v,u);
		sz[u]+=sz[v];
		if(sz[v]>sz[son[u]]) son[u]=v;
	}
}
void dfs2(int u,int t){
	dfn[u]=++timestamp,nw[timestamp]=w[u],top[u]=t;
//	printf("%d %d %d\n",u,dfn[u],top[u]);
	if(son[u]) dfs2(son[u],t);
	for(int i=h[u];~i;i=ne[i]){
		int v=e[i];
		if(v==fa[u]||v==son[u]) continue;
		dfs2(v,v);
	} 
}
struct matrix{
	int a[3][3];
	matrix(){
		for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
				a[i][j]=-INF;
	}
	void init(){
		for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
				a[i][j]=(i!=j)*(-INF);
	}
	matrix operator*(const matrix &W)const{
		matrix res;
		for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
				for(int k=0;k<3;k++)
					res.a[i][k]=max(res.a[i][k],a[i][j]+W.a[j][k]);
		return res;
	}
};
struct node{
	int l,r,same;
	matrix a,b;
}tr[N*4];
void init(matrix &x,int w,int cnt=1){
	x.a[0][0]=x.a[2][2]=0;
	x.a[1][1]=w*cnt;
	x.a[1][0]=x.a[2][0]=x.a[2][1]=w>=0?w*cnt:w;
	x.a[0][1]=x.a[0][2]=x.a[1][2]=-INF;
}
void pushup(node &t,node l,node r){
	t.a=l.a*r.a;
	t.b=r.b*l.b;
}
void pushup(int u){
	pushup(tr[u],tr[u<<1],tr[u<<1|1]);
}
void pushdown(node &t,int w){
	init(t.a,w,t.r-t.l+1),init(t.b,w,t.r-t.l+1);
	t.same=w;
} 
void pushdown(int u){
	if(tr[u].same!=-INF){
		pushdown(tr[u<<1],tr[u].same);
		pushdown(tr[u<<1|1],tr[u].same);
		tr[u].same=-INF;
	}
}
void build(int u,int l,int r){
//	printf("%d %d %d\n",u,l,r);
	tr[u]={l,r,-INF};
	if(l==r) init(tr[u].a,nw[r]),init(tr[u].b,nw[r]);
	else{
		int mid=l+r>>1;
		build(u<<1,l,mid),build(u<<1|1,mid+1,r);
		pushup(u);
	}
//	printf("--%d %d %d\n",u,l,r);
//	for(int i=0;i<3;i++,puts(""))
//		for(int j=0;j<3;j++) printf("%d ",tr[u].a.a[i][j]);
//	for(int i=0;i<3;i++,puts(""))
//		for(int j=0;j<3;j++) printf("%d ",tr[u].b.a[i][j]);
}
void modify(int u,int l,int r,int w){
	if(tr[u].l>=l&&tr[u].r<=r) pushdown(tr[u],w);
	else{
		pushdown(u);
		int mid=tr[u].l+tr[u].r>>1;
		if(l<=mid) modify(u<<1,l,r,w);
		else modify(u<<1|1,l,r,w);
		pushup(u);
	}
}
matrix query(int u,int l,int r,int t){
//	printf("------%d %d %d %d\n",u,l,r,t);
	if(tr[u].l>=l&&tr[u].r<=r){
		if(t==0) return tr[u].a;
		else return tr[u].b;
	}
	else{
		pushdown(u);
		int mid=tr[u].l+tr[u].r>>1;
		if(r<=mid) return query(u<<1,l,r,t);
		else if(l>mid) return query(u<<1|1,l,r,t);
		else{
			matrix left=query(u<<1,l,r,t),right=query(u<<1|1,l,r,t);
			return t==0?left*right:right*left;
		}
	}
}
void modify_path(int u,int v,int w){
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		modify(1,dfn[top[u]],dfn[u],w);
		u=fa[top[u]];
	}
	if(dep[u]<dep[v]) swap(u,v);
	modify(1,dfn[v],dfn[u],w); 
}
int query_path(int u,int v){
	matrix l,r; l.init(); r.init();
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]]){
			r=query(1,dfn[top[v]],dfn[v],0)*r;
			v=fa[top[v]];
		}
		else{
			l=l*query(1,dfn[top[u]],dfn[u],1);
			u=fa[top[u]];
		}
//		printf("%d %d\n",u,v);
//		for(int i=0;i<3;i++,puts(""))
//			for(int j=0;j<3;j++) printf("%d ",l.a[i][j]);
//		for(int i=0;i<3;i++,puts(""))
//			for(int j=0;j<3;j++) printf("%d ",r.a[i][j]);
//3 1 -2 3 1 2 1 3 1 1 2 3
//5 1 -2 3 -4 5 1 2 1 3 2 4 2 5 1 1 2 3
	}
	if(dep[u]<dep[v]) r=query(1,dfn[u],dfn[v],0)*r;
	else l=l*query(1,dfn[v],dfn[u],1);
//	for(int i=0;i<3;i++,puts(""))
//		for(int j=0;j<3;j++) printf("%d ",l.a[i][j]);
//	for(int i=0;i<3;i++,puts(""))
//		for(int j=0;j<3;j++) printf("%d ",r.a[i][j]);
	matrix res; res=l*r;
	return max(0,max(0,res.a[2][0]));
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&w[i]);
	memset(h,-1,sizeof(h));
	for(int i=1,a,b;i<n;i++){
		scanf("%d%d",&a,&b);
		add(a,b),add(b,a);
	}
	dfs1(1,0),dfs2(1,1);build(1,1,n);
	int op,a,b,c; scanf("%d",&Q);
	while(Q--){
		scanf("%d%d%d",&op,&a,&b);
		if(op==1) printf("%d\n",query_path(a,b));
		else{
			scanf("%d",&c);
			modify_path(a,b,c);
		}
	}
	return 0;
}
2023/7/16 21:17
加载中...