Description
一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。
我们将以下面的形式来要求你对这棵树完成一些操作:
I. CHANGE u t : 把结点u的权值改为t
II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值
III. QSUM u v: 询问从点u到点v的路径上的节点的权值和
注意:从点u到点v的路径上的节点包括u和v本身
我们将以下面的形式来要求你对这棵树完成一些操作:
I. CHANGE u t : 把结点u的权值改为t
II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值
III. QSUM u v: 询问从点u到点v的路径上的节点的权值和
注意:从点u到点v的路径上的节点包括u和v本身
Input
输入文件的第一行为一个整数n,表示节点的个数。
接下来n – 1行,每行2个整数a和b,表示节点a和节点b之间有一条边相连。
接下来n行,每行一个整数,第i行的整数wi表示节点i的权值。
接下来1行,为一个整数q,表示操作的总数。
接下来q行,每行一个操作,以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。
保证1<=n<=30000,0<=q<=200000;中途操作中保证每个节点的权值w在-30000到30000之间。
接下来n – 1行,每行2个整数a和b,表示节点a和节点b之间有一条边相连。
接下来n行,每行一个整数,第i行的整数wi表示节点i的权值。
接下来1行,为一个整数q,表示操作的总数。
接下来q行,每行一个操作,以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。
保证1<=n<=30000,0<=q<=200000;中途操作中保证每个节点的权值w在-30000到30000之间。
Output
对于每个“QMAX”或者“QSUM”的操作,每行输出一个整数表示要求输出的结果。
Sample Input
4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4
Sample Output
4
1
2
2
10
6
5
6
5
16
现在很晚了,这种板子题我也不知道怎么表述。
到时候再补吧,先丢代码
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=60001;
int n,m,cnt,now,son[N],f[N],nex[N],h[N],top[N],id[N],size[N],dep[N],ans,mxx;
struct oo{int a,b,v,sum,mx;}s[N*2-20000];
void ins(int x,int y)
{
son[++cnt]=y;
nex[cnt]=h[x];
h[x]=cnt;
}
void dfs(int x)
{
size[x]=1;
for(int i=h[x];i;i=nex[i])
{
int v=son[i];
if(v==f[x])continue;
dep[v]=dep[x]+1;
f[v]=x;
dfs(v);
size[x]+=size[v];
}
}
void dfs2(int x,int y)
{
int k=0;
id[x]=++now;top[x]=y;
for(int i=h[x];i;i=nex[i])
if(dep[son[i]]>dep[x]&&size[son[i]]>size[k])
k=son[i];
if(k==0)return ;
dfs2(k,y);
for(int i=h[x];i;i=nex[i])
{
if(son[i]==k)continue;
if(dep[son[i]]>dep[x])
dfs2(son[i],son[i]);
}
}
void build(int x,int l,int r)
{
s[x].a=l,s[x].b=r;
if(l==r){s[x].mx=-9999999;return ;}
build(x<<1,l,l+r>>1);
build(x<<1|1,(l+r>>1)+1,r);
}
void change(int x,int l,int v)
{
if(s[x].a==s[x].b){s[x].v=v;s[x].sum=v;s[x].mx=v;return ;}
int mid=s[x].a+s[x].b>>1;
if(l<=mid)change(x<<1,l,v);
else change(x<<1|1,l,v);
s[x].mx=max(s[x<<1].mx,s[x<<1|1].mx);
s[x].sum=s[x<<1].sum+s[x<<1|1].sum;
}
void get(int x,int l,int r)
{
if(l<=s[x].a&&r>=s[x].b)
ans+=s[x].sum,mxx=max(mxx,s[x].mx);
else
{
int mid=s[x].a+s[x].b>>1;
if(l<=mid)get(x<<1,l,r);
if(r>mid)get(x<<1|1,l,r);
}
}
void qmax(int x,int y)
{
int mx=-99999999;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
mxx=-99999999;ans=0;
get(1,id[top[x]],id[x]);
mx=max(mx,mxx);
x=f[top[x]];
}
if(id[x]>id[y])swap(x,y);
mxx=-99999999;ans=0;
get(1,id[x],id[y]);
mx=max(mx,mxx);
printf("%d\n",mx);
}
void qsum(int x,int y)
{
int sum=0;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
ans=0;
get(1,id[top[x]],id[x]);
sum+=ans;
x=f[top[x]];
}
if(id[x]>id[y])swap(x,y);
ans=0;
get(1,id[x],id[y]);
sum+=ans;
printf("%d\n",sum);
}
int main()
{
scanf("%d",&n);
for(int i=1,x,y;i<n;i++)
scanf("%d%d",&x,&y),ins(x,y),ins(y,x);
dfs(1),dfs2(1,1);build(1,1,n);
for(int i=1,x;i<=n;i++)
scanf("%d",&x),change(1,id[i],x);
scanf("%d",&m);char p[13];int a,b;
while(m--)
{
scanf("%s",p+1);
if(p[2]=='S')
{
scanf("%d%d",&a,&b);
qsum(a,b);
}
if(p[2]=='M')
{
scanf("%d%d",&a,&b);
qmax(a,b);
}
if(p[2]=='H')
{
scanf("%d%d",&a,&b);
change(1,id[a],b);
}
}
}
祝大家早日AC!!!