题目链接:点击查看
题意:给出n个点,每条边有边权,两种操作,1、查询路径上最大的权值 2、修改某边的权值
题解:把每条边赋给儿子,然后就是基本操作了。要注意的是,修改边的时候给出的是第几条边,因此输入的时候要保存一下
#include <bits/stdc++.h>
using namespace std;
const int N=1e4+10;
struct edge{
int to,nex,d;
}e[N*2];
int head[N],len;
int fa[N],deep[N],son[N],num[N],top[N],val[N];
int id[N],pid[N],cnt;
int posl[N],posr[N],posval[N];
int n;
struct node{
int l,r;
int val;
}tree[N<<2];
void init()
{
for(int i=1;i<=n;i++)
{
head[i]=-1;
son[i]=-1;
fa[i]=-1;
}
cnt=0;
len=0;
}
void addedge(int u,int v,int d)
{
e[len].to=v;
e[len].d=d;
e[len].nex=head[u];
head[u]=len++;
}
void dfs1(int u,int f)
{
num[u]=1;
int to;
for(int i=head[u];i!=-1;i=e[i].nex)
{
to=e[i].to;
if(to==f) continue;
val[to]=e[i].d;
fa[to]=u;
deep[to]=deep[u]+1;
dfs1(to,u);
if(son[u]==-1||num[to]>num[son[u]])
son[u]=to;
}
}
void dfs2(int u,int topfa)
{
top[u]=topfa;
id[u]=++cnt; pid[cnt]=u;
if(son[u]==-1) return;
dfs2(son[u],topfa);
int to;
for(int i=head[u];i!=-1;i=e[i].nex)
{
to=e[i].to;
if(to==fa[u]||to==son[u]) continue;
dfs2(to,to);
}
}
void pushup(int cur)
{
tree[cur].val=max(tree[cur<<1].val,tree[cur<<1|1].val);
}
void build(int l,int r,int cur)
{
tree[cur].l=l;
tree[cur].r=r;
if(l==r)
{
tree[cur].val=val[pid[l]];
return;
}
int mid=(r+l)>>1;
build(l,mid,cur<<1);
build(mid+1,r,cur<<1|1);
pushup(cur);
}
void update(int pos,int cur,int val)
{
if(tree[cur].l==tree[cur].r)
{
tree[cur].val=val;
return;
}
if(pos<=tree[cur<<1].r) update(pos,cur<<1,val);
else update(pos,cur<<1|1,val);
pushup(cur);
// cout<<tree[cur].val<<endl;
}
int query(int pl,int pr,int cur)
{
if(pl<=tree[cur].l&&tree[cur].r<=pr) return tree[cur].val;
int res=-1e9;
if(pl<=tree[cur<<1].r) res=max(res,query(pl,pr,cur<<1));
if(pr>=tree[cur<<1|1].l) res=max(res,query(pl,pr,cur<<1|1));
return res;
}
int solve(int l,int r)
{
int f1=top[l],f2=top[r];
int res=-1e9;
while(f1!=f2)
{
if(deep[f1]<deep[f2])
{
swap(f1,f2);
swap(l,r);
}
res=max(res,query(id[f1],id[l],1));
l=fa[f1];
f1=top[l];
}
if(deep[l]>deep[r]) swap(l,r);
// cout<<id[l]<<" "<<id[r]<<endl;
res=max(res,query(id[l]+1,id[r],1));
return res;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
init();
int u,v,d;
char op[10];
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&u,&v,&d);
posl[i]=u,posr[i]=v,posval[i]=d;
addedge(u,v,d);
addedge(v,u,d);
}
dfs1(1,-1);
dfs2(1,1);
build(1,n,1);
while(scanf("%s",op)&&strcmp(op,"DONE")!=0)
{
scanf("%d%d",&u,&v);
if(op[0]=='Q') printf("%d\n",solve(u,v));
else {
if(fa[posl[u]]==posr[u]) u=posl[u];
else u=posr[u];
update(id[u],1,v);
}
}
}
return 0;
}