only AC on #11
查看原帖
only AC on #11
763878
Jerry_heng楼主2024/1/5 21:46
#include<bits/stdc++.h>
using namespace std;
const int mxn=4e6+10;
struct node{
	int to,nxt,val;
}edge[mxn];
struct nodd{
	int x,y;
}id[mxn];
int q,cnt,f[mxn],cnt2,dfn[mxn],son[mxn],n,v[mxn],val[mxn],siz[mxn],head[mxn],tp[mxn],s[mxn],mn[mxn],mx[mxn],dep[mxn];
bool lz[mxn];
string op;
void add(int u,int v,int w){
	edge[++cnt]=(node){v,head[u],w};
	head[u]=cnt;
}
void dfs1(int now,int fa){
	f[now]=fa,dep[now]=dep[fa]+1,siz[now]=1;
	for(int i=head[now];i;i=edge[i].nxt){
		int v=edge[i].to;
		if(v==fa)continue;
		val[v]=edge[i].val;
		dfs1(v,now);
		siz[now]+=siz[v];
		if(siz[v]>siz[son[now]])son[now]=v;
	}
}
void dfs2(int now,int top){
	tp[now]=top;
	dfn[now]=++cnt2;
	v[cnt2]=val[now];
	if(son[now])dfs2(son[now],top);
	for(int i=head[now];i;i=edge[i].nxt){
		int v=edge[i].to;
		if(v==f[now]||v==son[now])continue;
		dfs2(v,v);
	}
}
void pushup(int o){
	s[o]=s[o*2]+s[o*2+1];
	mn[o]=min(mn[o*2],mn[o*2+1]);
	mx[o]=max(mx[o*2],mx[o*2+1]);
}
void pushdown(int o){
	lz[o*2]^=1,lz[o*2+1]^=1;
	s[o*2]=-s[o*2],s[o*2+1]=-s[o*2+1];
	mn[o*2]=-mn[o*2],mx[o*2]=-mx[o*2];
	mn[o*2+1]=-mn[o*2+1],mx[o*2+1]=-mx[o*2+1];
	swap(mn[o*2],mx[o*2]),swap(mn[o*2+1],mx[o*2+1]);
	lz[o]^=1;
}
void build(int o,int l,int r){
	if(l==r){
		s[o]=mn[o]=mx[o]=v[l];
		return;
	}
	int mid=(l+r)>>1;
	build(o*2,l,mid);
	build(o*2+1,mid+1,r);
	pushup(o); 
}
void update(int o,int l,int r,int pos,int w){
	if(l==r){
		s[o]=mn[o]=mx[o]=w;
		return;
	}
	int mid=(l+r)>>1;
	if(pos<=mid)update(o*2,l,mid,pos,w);
	else update(o*2+1,mid+1,r,pos,w);
	pushup(o);
}
void change(int o,int l,int r,int L,int R){
	if(L<=l&&r<=R){
		lz[o]^=1;
		s[o]=-s[o];
		mn[o]=-mn[o];
		mx[o]=-mx[o];
		swap(mn[o],mx[o]);
		return;
	} 
	int mid=(l+r)>>1;
	if(lz[o])pushdown(o);
	if(L<=mid)change(o*2,l,mid,L,R);
	if(R>mid)change(o*2+1,mid+1,r,L,R);
	pushup(o);
}
void chang(int u,int v){
	while(tp[u]!=tp[v]){
		if(dep[tp[u]]<dep[tp[v]])swap(u,v);
		change(1,1,n,dfn[tp[u]],dfn[u]);
		u=f[tp[u]];
	}
	if(dep[u]>dep[v])swap(u,v);
	if(u!=v)change(1,1,n,dfn[u]+1,dfn[v]);
}
int querys(int o,int l,int r,int L,int R){
	if(L<=l&&r<=R)return s[o];
	if(lz[o])pushdown(o);
	int mid=(l+r)>>1,ans=0;
	if(L<=mid)ans+=querys(o*2,l,mid,L,R);
	if(R>mid)ans+=querys(o*2+1,mid+1,r,L,R);
	pushup(o);
	return ans;
}
int querymx(int o,int l,int r,int L,int R){
	if(L<=l&&r<=R)return mx[o];
	if(lz[o])pushdown(o);
	int mid=(l+r)>>1,ans=-INT_MAX;
	if(L<=mid)ans=max(ans,querymx(o*2,l,mid,L,R));
	if(R>mid)ans=max(ans,querymx(o*2+1,mid+1,r,L,R));
	pushup(o);
	return ans;
}
int querymn(int o,int l,int r,int L,int R){
	if(L<=l&&r<=R)return mn[o];
	if(lz[o])pushdown(o);
	int mid=(l+r)>>1,ans=INT_MAX;
	if(L<=mid)ans=min(ans,querymn(o*2,l,mid,L,R));
	if(R>mid)ans=min(ans,querymn(o*2+1,mid+1,r,L,R));
	pushup(o);
	return ans;
}
int qs(int u,int v){
	int ans=0;
	while(tp[u]!=tp[v]){
		if(dep[tp[u]]<dep[tp[v]])swap(u,v);
		ans+=querys(1,1,n,dfn[tp[u]],dfn[u]);
		u=f[tp[u]];
	}
	if(dep[u]>dep[v])swap(u,v);
	if(u!=v)ans+=querys(1,1,n,dfn[u]+1,dfn[v]);
	return ans;
}
int qmx(int u,int v){
	int ans=-INT_MAX;
	while(tp[u]!=tp[v]){
		if(dep[tp[u]]<dep[tp[v]])swap(u,v);
		ans=max(ans,querymx(1,1,n,dfn[tp[u]],dfn[u]));
		u=f[tp[u]];
	}
	if(dep[u]>dep[v])swap(u,v);
	if(u!=v)ans=max(ans,querymx(1,1,n,dfn[u]+1,dfn[v]));
	return ans;
}
int qmn(int u,int v){
	int ans=INT_MAX;
	while(tp[u]!=tp[v]){
		if(dep[tp[u]]<dep[tp[v]])swap(u,v);
		ans=min(ans,querymn(1,1,n,dfn[tp[u]],dfn[u]));
		u=f[tp[u]];
	}
	if(dep[u]>dep[v])swap(u,v);
	if(u!=v)ans=min(ans,querymn(1,1,n,dfn[u]+1,dfn[v]));
	return ans;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u+1,v+1,w);add(v+1,u+1,w);
		id[i].x=u+1,id[i].y=v+1;
	}
	dfs1(1,0);dfs2(1,1);build(1,1,n);
	scanf("%d",&q);       
	while(q--){
		cin>>op;
		if(op=="C"){
			int u,w,uu;
			scanf("%d%d",&u,&w);
			if(dep[id[u].x]<dep[id[u].y])uu=id[u].y;
			else uu=id[u].x;
			update(1,1,n,dfn[uu],w);
		}
		if(op=="N"){
			int u,v;
			scanf("%d%d",&u,&v);
			u++,v++;
			chang(u,v);
		}
		if(op=="SUM"){
			int u,v;
			scanf("%d%d",&u,&v);
			u++,v++;
			printf("%d\n",qs(u,v));
		}
		if(op=="MAX"){
			int u,v;
			scanf("%d%d",&u,&v);
			u++,v++;
			printf("%d\n",qmx(u,v));
		}
		if(op=="MIN"){
			int u,v;
			scanf("%d%d",&u,&v);
			u++,v++;
			printf("%d\n",qmn(u,v));
		}
	}
	return 0;
}
2024/1/5 21:46
加载中...