8pts,AC#11
#include<cstdio>
#define srz x++,y++
int max(int a,int b)
{
return a>b?a:b;
}
int min(int a,int b)
{
return a<b?a:b;
}
void swap(int &a,int &b)
{
int tmp=a;
a=b;
b=tmp;
}
const int N=2e5+10;
struct edge
{
int v;
int w;
int nxt;
}e[N<<1];
int head[N],tot;
int from[N],to[N];
void add(int u,int v,int w)
{
e[++tot].v=v;
e[tot].w=w;
e[tot].nxt=head[u];
head[u]=tot;
}
int n,m;
int a[N];
int fa[N],dep[N],size[N],son[N];
void dfs1(int u,int f)
{
size[u]=1;
for(int i=head[u];i;i=e[i].nxt)
{
int v=e[i].v;
int w=e[i].w;
if(v==f) continue;
dep[v]=dep[u]+1;
fa[v]=u;
a[v]=w;
dfs1(v,u);
size[u]+=size[v];
if(size[son[u]]<size[v]) son[u]=v;
}
}
int id[N],top[N],rev[N],total;
void dfs2(int u,int t)
{
top[u]=t;
id[u]=++total;
rev[total]=a[u];
if(!son[u]) return ;
dfs2(son[u],t);
for(int i=head[u];i;i=e[i].nxt)
{
int v=e[i].v;
if(v==fa[u]||v==son[u]) continue;
dfs2(v,v);
}
}
struct seg
{
struct node
{
int l;
int r;
int maxn;
int minn;
int sum;
int lz;
}t[N<<2];
#define ls k<<1
#define rs k<<1|1
#define mid ((t[k].l+t[k].r)>>1)
void pushup(int k)
{
t[k].sum=t[ls].sum+t[rs].sum;
t[k].maxn=max(t[ls].maxn,t[rs].maxn);
t[k].minn=min(t[ls].minn,t[rs].minn);
}
void pushdown(int k)
{
if(!t[k].lz) return ;
t[ls].lz^=1;
t[rs].lz^=1;
t[ls].sum=-t[ls].sum;
t[rs].sum=-t[rs].sum;
t[ls].maxn=-t[ls].maxn;
t[rs].maxn=-t[rs].maxn;
t[ls].minn=-t[ls].minn;
t[rs].minn=-t[rs].minn;
swap(t[ls].maxn,t[ls].minn);
swap(t[rs].maxn,t[rs].minn);
t[ls].lz=0;
}
void build(int k,int l,int r)
{
t[k].l=l,t[k].r=r;
if(l==r)
{
t[k].maxn=t[k].minn=t[k].sum=rev[l];
return ;
}
build(ls,l,mid);
build(rs,mid+1,r);
pushup(k);
}
void update_c(int k,int x,int num)
{
if(t[k].l==t[k].r)
{
t[k].maxn=t[k].minn=t[k].sum=num;
return ;
}
pushdown(k);
if(x<=mid) update_c(ls,x,num);
else update_c(rs,x,num);
pushup(k);
}
void update_n(int k,int l,int r)
{
if(t[k].l>=l&&t[k].r<=r)
{
t[k].lz^=1;
t[k].sum=-t[k].sum;
t[k].maxn=-t[k].maxn;
t[k].minn=-t[k].minn;
swap(t[k].maxn,t[k].minn);
return ;
}
pushdown(k);
if(l<=mid) update_n(ls,l,r);
if(r>mid) update_n(rs,l,r);
pushup(k);
}
int query_sum(int k,int l,int r)
{
if(t[k].l>=l&&t[k].r<=r) return t[k].sum;
pushdown(k);
int ans=0;
if(l<=mid) ans+=query_sum(ls,l,r);
if(r>mid) ans+=query_sum(rs,l,r);
return ans;
}
int query_maxn(int k,int l,int r)
{
if(t[k].l>=l&&t[k].r<=r) return t[k].maxn;
pushdown(k);
int ans=-114514;
if(l<=mid) ans=max(ans,query_maxn(ls,l,r));
if(r>mid) ans=max(ans,query_maxn(rs,l,r));
return ans;
}
int query_minn(int k,int l,int r)
{
if(t[k].l>=l&&t[k].r<=r) return t[k].minn;
pushdown(k);
int ans=114514;
if(l<=mid) ans=min(ans,query_minn(ls,l,r));
if(r>mid) ans=min(ans,query_minn(rs,l,r));
return ans;
}
}tree;
void upd_pp(int x,int y)
{
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]]) swap(x,y);
tree.update_n(1,id[top[x]],id[x]);
x=fa[top[x]];
}
if(dep[x]>dep[y]) swap(x,y);
tree.update_n(1,id[x]+1,id[y]);
}
int qsum(int x,int y)
{
int ans=0;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]]) swap(x,y);
ans+=tree.query_sum(1,id[top[x]],id[x]);
x=fa[top[x]];
}
if(dep[x]>dep[y]) swap(x,y);
ans+=tree.query_sum(1,id[x]+1,id[y]);
return ans;
}
int qmax(int x,int y)
{
int ans=-114514;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]]) swap(x,y);
ans=max(ans,tree.query_maxn(1,id[top[x]],id[x]));
x=fa[top[x]];
}
if(dep[x]>dep[y]) swap(x,y);
ans=max(ans,tree.query_maxn(1,id[x]+1,id[y]));
return ans;
}
int qmin(int x,int y)
{
int ans=114514;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]]) swap(x,y);
ans=min(ans,tree.query_minn(1,id[top[x]],id[x]));
x=fa[top[x]];
}
if(dep[x]>dep[y]) swap(x,y);
ans=min(ans,tree.query_minn(1,id[x]+1,id[y]));
return ans;
}
int main()
{
scanf("%d",&n);
for(int i=1,u,v,w;i<=n-1;i++)
{
scanf("%d%d%d",&u,&v,&w);
u++;
v++;
add(u,v,w);
add(v,u,w);
from[i]=u,to[i]=v;
}
dfs1(1,0);
dfs2(1,1);
tree.build(1,1,n);
scanf("%d",&m);
while(m--)
{
char op[10];
int x,y;
scanf(" %s %d%d",op,&x,&y);
if(op[0]=='C')
{
int u=(dep[from[x]]>dep[to[x]])?from[x]:to[x];
tree.update_c(1,id[u],y);
}
else if(op[0]=='N') srz,upd_pp(x,y);
else if(op[0]=='S') srz,printf("%d\n",qsum(x,y));
else if(op[1]=='A') srz,printf("%d\n",qmax(x,y));
else srz,printf("%d\n",qmin(x,y));
}
return 0;
}
另:遇到树剖题有什么调试的策略吗?
(现状:写一道树剖WA大半天