總算是A了一道鏈剖的題目了/(ㄒoㄒ)/~~。
雖然說一開始也是GG了
題目
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
解法
裸的樹鏈剖分,很裸很裸。。
注意負數的情況,當時被坑了很久(其實也就是幾分鐘)
代碼
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=1e4*3+5;
int fre[maxn],go[maxn*2],next[maxn*2],a[maxn],top[maxn],id[maxn],deep[maxn],size[maxn];
int son[maxn],n,q,now[maxn],tot,num,ans,dad[maxn];
struct cy{
int mx,sum;
}tree[maxn*4];
void add(int x,int y)
{
go[++num]=y;
next[num]=fre[x];
fre[x]=num;
}
void dfs1(int x,int fa,int dep)
{
deep[x]=dep; dad[x]=fa; size[x]=1;
int big=0;
for(int i=fre[x];i;i=next[i]){
if (go[i]==fa) continue;
dfs1(go[i],x,dep+1);
if (size[go[i]]>big) big=size[go[i]],son[x]=go[i];
size[x]+=size[go[i]];
}
}
void dfs2(int x,int tp)
{
id[x]=++tot; top[x]=tp; now[tot]=a[x];
if (son[x]!=-1) dfs2(son[x],tp);
for(int i=fre[x];i;i=next[i]){
if (go[i]==dad[x]) continue;
if (go[i]==son[x]) continue;
dfs2(go[i],go[i]);
}
}
void maketree(int p,int l,int r)
{
if (l==r) {
tree[p].mx=now[l];
tree[p].sum=now[l];
return;
}
int mid=(l+r)>>1;
maketree(p<<1,l,mid);
maketree(p<<1|1,mid+1,r);
tree[p].mx=max(tree[p<<1].mx,tree[p<<1|1].mx);
tree[p].sum=tree[p<<1].sum+tree[p<<1|1].sum;
}
void change(int p,int l,int r,int pos,int val)
{
if (l==r) tree[p].mx=val,tree[p].sum=val;
else {
int mid=(l+r)>>1;
if (pos<=mid) change(p<<1,l,mid,pos,val);
else change(p<<1|1,mid+1,r,pos,val);
tree[p].mx=max(tree[p<<1].mx,tree[p<<1|1].mx);
tree[p].sum=tree[p<<1].sum+tree[p<<1|1].sum;
}
}
void getmax(int p,int l,int r,int a,int b)
{
if (l==a&&r==b) ans=max(ans,tree[p].mx);
else {
int mid=(l+r)>>1;
if (b<=mid) getmax(p<<1,l,mid,a,b);
else if (a>mid) getmax(p<<1|1,mid+1,r,a,b);
else {
getmax(p<<1,l,mid,a,mid);
getmax(p<<1|1,mid+1,r,mid+1,b);
}
}
}
void getsum(int p,int l,int r,int a,int b)
{
if (l==a&&r==b) ans+=tree[p].sum;
else {
int mid=(l+r)>>1;
if (b<=mid) getsum(p<<1,l,mid,a,b);
else if (a>mid) getsum(p<<1|1,mid+1,r,a,b);
else {
getsum(p<<1,l,mid,a,mid);
getsum(p<<1|1,mid+1,r,mid+1,b);
}
}
}
void find(int x,int y,int pd)
{
int tpx=top[x],tpy=top[y];
while (tpx!=tpy){
if (deep[tpx]<deep[tpy]){
swap(tpx,tpy);
swap(x,y);
}
if (!pd) getmax(1,1,tot,id[tpx],id[x]);
else getsum(1,1,tot,id[tpx],id[x]);
x=dad[tpx],tpx=top[x];
}
if (deep[x]>deep[y]) swap(x,y);
if (!pd) getmax(1,1,tot,id[x],id[y]);
else getsum(1,1,tot,id[x],id[y]);
}
int main()
{
// freopen("T.in","r",stdin);
scanf("%d",&n);
fo(i,1,n-1){
int x,y;
scanf("%d%d",&x,&y);
add(x,y); add(y,x);
}
fo(i,1,n) scanf("%d",&a[i]);
memset(son,255,sizeof(son));
dfs1(1,0,1);
dfs2(1,1);
maketree(1,1,tot);
scanf("%d",&q);
fo(i,1,q){
char ch[10];
int x,y;
scanf("%s%d%d",&ch,&x,&y);
if (ch[0]=='C') change(1,1,tot,id[x],y);
else if (ch[1]=='M') ans=-1e9,find(x,y,0),printf("%d\n",ans);
else ans=0,find(x,y,1),printf("%d\n",ans);
}
}