洛谷 P1505 [国家集训队]旅游 题解

题目

此题的关键，在于维护区间取反的懒标记，以及取反后区间最大值，最小值与和的变化
首先，我们肯定先要把边权转化为点权。这个相信大家都会，每个点权等于他返祖边的权值

设 $maxv[k],minv[k],sum[k]$ 分别为区间 $k$ 的最大值，最小值与和；$rev[k]=0/1$ 表示该区间是否取反。
取反后 $maxv[k],minv[k],sum[k]$ 的变化：

• $sum[k]$ 最简单的一个，直接乘以 $-1$ 即可
• $minv[k]$ 等于取反前的 $maxv[k]\times -1$，可以根据不等式的运算法则得到
• $maxv[k]$ 等于取反前的 $minv[k]\times -1$ ，与上同理

这个地方明白后，剩下的就是树剖的板子了

#include<cstdio>
#include<iostream>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;
const int Maxn=100000+10,Maxm=400000+10,inf=0x3f3f3f3f;
struct edge{
	int v,len;
	edge(int x,int y)
	{
		v=x,len=y;
	}
};
struct Edge{
	int u,v;
}g[Maxn];
int maxv[Maxm],minv[Maxm],sum[Maxm];
bool rev[Maxn];
int a[Maxn],w[Maxn];
int d[Maxn],top[Maxn],id[Maxn];
int s[Maxn],son[Maxn],f[Maxn];
vector <edge> e[Maxn];
int n,m,idcnt;
inline int read()
{
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
	return s*w;
}
inline void push_up(int k)
{
	maxv[k]=max(maxv[k<<1],maxv[k<<1|1]);
	minv[k]=min(minv[k<<1],minv[k<<1|1]);
	sum[k]=sum[k<<1]+sum[k<<1|1];
}
void upd(int k)
{
	rev[k]^=1;
	sum[k]=-sum[k];
	int tmp=minv[k];
	minv[k]=-maxv[k];
	maxv[k]=-tmp;
}
void push_down(int k)
{
	if(!rev[k])return;
	upd(k<<1),upd(k<<1|1);
	rev[k]=0;
}
void modify_pos(int k,int l,int r,int pos,int v)
{
	if(l==r)
	{
		rev[k]=0;
		maxv[k]=minv[k]=sum[k]=v;
		return;
	}
	push_down(k);
	int mid=(l+r)>>1;
	if(pos<=mid)modify_pos(k<<1,l,mid,pos,v);
	else modify_pos(k<<1|1,mid+1,r,pos,v);
	push_up(k);
}
void modify_rev(int k,int l,int r,int x,int y)
{
	if(x<=l && r<=y)return upd(k);
	push_down(k);
	int mid=(l+r)>>1;
	if(x<=mid)modify_rev(k<<1,l,mid,x,y);
	if(mid<y)modify_rev(k<<1|1,mid+1,r,x,y);
	push_up(k);
}
int query_sum(int k,int l,int r,int x,int y)
{
	if(x<=l && r<=y)return sum[k];
	push_down(k);
	int mid=(l+r)>>1,ret=0;
	if(x<=mid)ret=query_sum(k<<1,l,mid,x,y);
	if(mid<y)ret+=query_sum(k<<1|1,mid+1,r,x,y);
	return ret;
}
int query_min(int k,int l,int r,int x,int y)
{
	if(x<=l && r<=y)return minv[k];
	push_down(k);
	int mid=(l+r)>>1,ret=inf;
	if(x<=mid)ret=query_min(k<<1,l,mid,x,y);
	if(mid<y)ret=min(ret,query_min(k<<1|1,mid+1,r,x,y));
	return ret;
}
int query_max(int k,int l,int r,int x,int y)
{
	if(x<=l && r<=y)return maxv[k];
	push_down(k);
	int mid=(l+r)>>1,ret=-inf;
	if(x<=mid)ret=query_max(k<<1,l,mid,x,y);
	if(mid<y)ret=max(ret,query_max(k<<1|1,mid+1,r,x,y));
	return ret;
}
void build(int k,int l,int r)
{
	if(l==r)
	{
		sum[k]=minv[k]=maxv[k]=w[l];
		return;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	push_up(k);
}
void dfs1(int x,int fa,int len)
{
	s[x]=1,f[x]=fa;
	d[x]=d[fa]+1,a[x]=len;
	for(int i=0;i<e[x].size();++i)
	{
		int y=e[x][i].v;
		if(y==fa)continue;
		dfs1(y,x,e[x][i].len);
		s[x]+=s[y];
		if(s[y]>s[son[x]])son[x]=y;
	}
}
void dfs2(int x,int topp)
{
	top[x]=topp,id[x]=++idcnt;
	w[id[x]]=a[x];
	if(!son[x])return;
	dfs2(son[x],topp);
	for(int i=0;i<e[x].size();++i)
	{
		int y=e[x][i].v;
		if(y==f[x] || y==son[x])continue;
		dfs2(y,y);
	}
}
void seq_rev(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(d[top[x]]<d[top[y]])swap(x,y);
		modify_rev(1,1,n,id[top[x]],id[x]);
		x=f[top[x]];
	}
	if(x==y)return;
	if(d[x]>d[y])swap(x,y);
	modify_rev(1,1,n,id[x]+1,id[y]);
}
int seq_sum(int x,int y) 
{
	int ret=0;
	while(top[x]!=top[y])
	{
		if(d[top[x]]<d[top[y]])swap(x,y);
		ret+=query_sum(1,1,n,id[top[x]],id[x]);
		x=f[top[x]];
	}
	if(x==y)return ret;
	if(d[x]>d[y])swap(x,y);
	return ret+query_sum(1,1,n,id[x]+1,id[y]);
}
int seq_max(int x,int y) 
{
	int ret=-inf;
	while(top[x]!=top[y])
	{
		if(d[top[x]]<d[top[y]])swap(x,y);
		ret=max(ret,query_max(1,1,n,id[top[x]],id[x]));
		x=f[top[x]];
	}
	if(x==y)return ret;
	if(d[x]>d[y])swap(x,y);
	return max(ret,query_max(1,1,n,id[x]+1,id[y]));
}
int seq_min(int x,int y) 
{
	int ret=inf;
	while(top[x]!=top[y])
	{
		if(d[top[x]]<d[top[y]])swap(x,y);
		ret=min(ret,query_min(1,1,n,id[top[x]],id[x]));
		x=f[top[x]];
	}
	if(x==y)return ret;
	if(d[x]>d[y])swap(x,y);
	return min(ret,query_min(1,1,n,id[x]+1,id[y]));
}
int main()
{
//	freopen("in.txt","r",stdin);
//	freopen("out.txt","w",stdout);
	
	n=read();
	for(int i=1;i<n;++i)
	{
		int x=read()+1,y=read()+1,c=read();
		g[i].u=x,g[i].v=y;
		e[x].push_back(edge(y,c));
		e[y].push_back(edge(x,c));
	}
	dfs1(1,0,0);
	dfs2(1,1);
	build(1,1,n);
	m=read();
	for(int i=1;i<=m;++i)
	{
		char opt[10];
		scanf("%s",opt);
		if(opt[0]=='C')
		{
			int i=read(),v=read();
			if(d[g[i].u]<d[g[i].v])swap(g[i].v,g[i].u);
			modify_pos(1,1,n,id[g[i].u],v);
			continue;
		}
		int x=read()+1,y=read()+1;
		if(opt[0]=='N')seq_rev(x,y);
		else if(opt[0]=='S')printf("%d\n",seq_sum(x,y));
		else if(opt[0]='M' && opt[1]=='A')printf("%d\n",seq_max(x,y));
		else if(opt[0]='M' && opt[1]=='I')printf("%d\n",seq_min(x,y));
	}
	
	return 0;
}