思路:树链剖分的又一个模板题,不过坑了好久...不知道为什么输入询问的时候用string会RE...
#include<iostream>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<string>
using namespace std;
const int maxn = 30005;
#define lson i<<1,l,m
#define rson i<<1|1,m+1,r
int n,tot;
int dep[maxn],son[maxn],siz[maxn],fa[maxn],top[maxn],id[maxn];
vector<int>e[maxn];
int val[maxn];
int maxv[maxn*4];
int sum[maxn*4];
int temp[maxn];
void push_up(int i)
{
maxv[i] = max(maxv[i<<1],maxv[i<<1|1]);
sum[i] = sum[i<<1]+sum[i<<1|1];
}
void build(int i,int l,int r)
{
maxv[i]=-0x3f3f3f3f;
if(l==r)
{
maxv[i]=val[l];
sum[i]=val[l];
return;
}
int m = (l+r)>>1;
build(lson);
build(rson);
push_up(i);
}
void update(int v,int vall,int i,int l,int r)
{
if(l==r)
{
sum[i] = vall;
maxv[i]=vall;
return;
}
int m = (l+r)>>1;
if(v <= m)
update(v,vall,lson);
else
update(v,vall,rson);
push_up(i);
}
int query1(int ql,int qr,int i,int l,int r)
{
if(ql<=l && qr>=r)
return sum[i];
int m = (l+r)>>1;
int ans = 0;
if(ql<=m)
ans+=query1(ql,qr,lson);
if(qr>m)
ans+=query1(ql,qr,rson);
return ans;
}
int query2(int ql,int qr,int i,int l,int r)
{
if(ql<=l && qr>=r)
return maxv[i];
int m = (l+r)>>1;
int ans = -1e9;
if(ql<=m)
ans = max(query2(ql,qr,lson),ans);
if(qr>m)
ans=max(query2(ql,qr,rson),ans);
return ans;
}
void dfs1(int u,int f,int d)
{
dep[u]=d;
fa[u]=f;
son[u]=0;
siz[u]=1;
for(int i = 0;i<e[u].size();i++)
{
int v = e[u][i];
if(v==f)
continue;
dfs1(v,u,d+1);
siz[u]+=siz[v];
if (siz[son[u]] < siz[v])
son[u]=v;
}
}
void dfs2(int u,int tp)
{
top[u]=tp;
id[u]=++tot;
if(son[u])
dfs2(son[u],tp);
for(int i = 0;i<e[u].size();i++)
{
int v = e[u][i];
if(v==fa[u]|| v==son[u])
continue;
dfs2(v,v);
}
}
int Yougth2(int u,int v)
{
int tp1 = top[u],tp2 = top[v];
int ans = -1e9;
while(tp1!=tp2)
{
if(dep[tp1] < dep[tp2])
{
swap(tp1,tp2);
swap(u,v);
}
ans = max(query2(id[tp1],id[u],1,1,tot),ans);
u = fa[tp1];
tp1 = top[u];
}
if(dep[u] > dep[v])
swap(u,v);
ans=max(query2(id[u],id[v],1,1,tot),ans);
return ans;
}
int Yougth1(int u,int v)
{
int tp1 = top[u],tp2 = top[v];
int ans = 0;
while(tp1!=tp2)
{
if(dep[tp1] < dep[tp2])
{
swap(tp1,tp2);
swap(u,v);
}
ans+=query1(id[tp1],id[u],1,1,tot);
u = fa[tp1];
tp1 = top[u];
}
if(dep[u] > dep[v])
swap(u,v);
ans+=query1(id[u],id[v],1,1,tot);
return ans;
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
tot = 0;
memset(val,0,sizeof(val));
memset(son,0,sizeof(son));
for(int i = 0;i<=n;i++)
e[i].clear();
for(int i =1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
e[u].push_back(v);
e[v].push_back(u);
}
for(int i = 1;i<=n;i++)
scanf("%d",&temp[i]);
dfs1(1,0,1);
dfs2(1,1);
for(int i = 1;i<=n;i++)
val[id[i]]=temp[i];
build(1,1,tot);
int q;
scanf("%d",&q);
while(q--)
{
char s[105];
int a,b;
scanf("%s%d%d",s,&a,&b);
if(s[1] == 'M')
{
printf("%d\n",Yougth2(a,b));
}
if(s[1]=='S')
{
printf("%d\n",Yougth1(a,b));
}
if (s[1]=='H')
{
update(id[a],b,1,1,tot);
}
}
}
}
Description
一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. 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”的形式给出。
对于100%的数据,保证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