树剖
只是想来放一波代码= =
单点修改+区间求和+区间最大值
区间修改就在线段树上加个Lazy操作就可以了
#include<iostream>
#include<cstdio>
#define N 300000
#define ll long long
using namespace std;
ll maxxx=0;
ll V[N];
ll Last[N],Next[N],End[N],tot;
ll Size[N],Fa[N],Dep[N],Son[N],Top[N],Dfn[N],Old[N],cnt;
//all of the arrays use old ids
struct node{ll l,r,v,Max;}tree[N];
void Find(ll u,ll uu,ll dep){
Size[u]=1,Fa[u]=uu,Dep[u]=dep;
for(ll i=Last[u];i;i=Next[i]){
ll v=End[i];
if(v==Fa[u])continue;//Go Wrong 1
Find(v,u,dep+1);
Size[u]+=Size[v];
if(Size[v]>Size[Son[u]])Son[u]=v;
}
}
void Connect(ll u,ll anc){
Top[u]=anc,Dfn[u]=++cnt,Old[cnt]=u;
if(Son[u])Connect(Son[u],anc);
for(ll i=Last[u];i;i=Next[i]){
ll v=End[i];
if(v!=Son[u] && v!=Fa[u])Connect(v,v);
}
}
void Make(ll p,ll x,ll y){
tree[p].l=x,tree[p].r=y;
if(x>=y){
tree[p].v=V[Old[x]];//Go Wrong 2:tree[p].v=V[x];
tree[p].Max=V[Old[x]];//Go Wrong 2
return;
}
ll mid=(x+y)>>1;
Make(p<<1,x,mid),Make((p<<1)+1,mid+1,y);
tree[p].v=tree[p<<1].v+tree[(p<<1)+1].v;
tree[p].Max=max(tree[p<<1].Max,tree[(p<<1)+1].Max);
}
void Modify(ll p,ll u,ll t){
if(tree[p].l==tree[p].r){
tree[p].v=t,tree[p].Max=t;
return;
}
int mid=(tree[p].l+tree[p].r)>>1;
if(u<=mid)Modify(p<<1,u,t);
else Modify((p<<1)+1,u,t);
tree[p].v=tree[p<<1].v+tree[(p<<1)+1].v;
tree[p].Max=max(tree[p<<1].Max,tree[(p<<1)+1].Max);
}
ll TreeSum(ll p,ll x,ll y){
if(tree[p].r<x || tree[p].l>y)return 0;
if(tree[p].l>=x && tree[p].r<=y)return tree[p].v;
return TreeSum(p<<1,x,y)+TreeSum((p<<1)+1,x,y);
}
ll GetSum(ll x,ll y){
ll Ans=0;
while(Top[x]!=Top[y]){
if(Dep[Top[x]]<Dep[Top[y]])swap(x,y);
Ans+=TreeSum(1,Dfn[Top[x]],Dfn[x]);
x=Fa[Top[x]];//Go Wrong 3
}
if(Dep[x]>Dep[y])swap(x,y);
Ans+=TreeSum(1,Dfn[x],Dfn[y]);
return Ans;
}
ll TreeMax(ll p,ll x,ll y){
if(tree[p].l>y || tree[p].r<x)return -1e9;
if(tree[p].l>=x && tree[p].r<=y)return tree[p].Max;
return max(TreeMax(p<<1,x,y),TreeMax((p<<1)+1,x,y));
}
ll GetMax(ll x,ll y){
ll Ans=-1e9;
while(Top[x]!=Top[y]){
if(Dep[Top[x]]<Dep[Top[y]])swap(x,y);
Ans=max(Ans,TreeMax(1,Dfn[Top[x]],Dfn[x]));
x=Fa[Top[x]];//Go Wrong 3
}
if(Dep[x]>Dep[y])swap(x,y);
Ans=max(Ans,TreeMax(1,Dfn[x],Dfn[y]));
return Ans;
}
void Ins(ll x,ll y){
End[++tot]=y;
Next[tot]=Last[x];
Last[x]=tot;
}
int main(){
ll n,q;scanf("%lld",&n);
for(ll i=1;i<n;i++){
ll a,b;scanf("%lld%lld",&a,&b);
Ins(a,b),Ins(b,a);
}
for(ll i=1;i<=n;i++)scanf("%lld",&V[i]);
Find(1,0,1);
Connect(1,1);
Make(1,1,n);
scanf("%lld",&q);
for(ll i=1;i<=q;i++){
char s[10];ll a,b;
scanf("%s%lld%lld",s,&a,&b);
if(s[1]=='H')Modify(1,Dfn[a],b);
if(s[1]=='M')printf("%lld\n",GetMax(a,b));
if(s[1]=='S')printf("%lld\n",GetSum(a,b));
}
return 0;
}
批注了Go Wrong的地方都是WA了之后才发现的错误之处。希望以后不会再在这些地方出错。
Something Goes Wrong !
Go Wrong 1
没有跳过v==father[u]的时候的情况。由于连的是无向边,所以在建立搜索树的时候必须判断。
Go Wrong 2
线段树里面每个点的编号都是搜索树里的新编号,不是题目里直接给的编号。在赋值的时候应该对应新编号。
Go Wrong 3
当操作完该条重链之后,x应该跳到这条链顶端的点的父亲节点,而不是跳到自己的父亲节点(这里指的都是搜索树里面的父子关系)。x=father[x]是标准的错误写法,保证又WA又TLE,一定是x=father[top[x]]。
LCA
void GetLCA(int x,int y){
while(Top[x]!=Top[y]){
if(Dep[Top[x]]<Dep[Top[y]])swap(x,y);
x=Fa[Top[x]];
}
if(Dep[x]>Dep[y])swap(x,y);
return x;
}