题意:有n个节点,每两个节点有一条唯一路径,有两种操作
1:给l到r的节点+k或者-k
2:查询x节点的值
思路:裸的树链剖分加一个线段树区间更新,网上有好多代码写得明显不对居然过了,这题数据太水了,还发表了博客,第一次写树链剖分,写下博客,当作笔记。
#include<bits/stdc++.h>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int maxn = 50005;
int a[maxn],rnk[maxn],siz[maxn],son[maxn],id[maxn],fa[maxn],top[maxn],dep[maxn],lazy[maxn*4],sum[maxn*4];
int topw;
int M,N,P;
vector<int> G[maxn];
void dfs_1(int u,int f,int d)
{
siz[u] = 1;
fa[u] = f;
dep[u] = d;
son[u] = 0;
for(int i=0;i<G[u].size();i++)
{
int v = G[u][i];
if(v == f) continue;
dfs_1(v,u,d+1);
siz[u] += siz[v];
if(siz[son[u]] < siz[v])
son[u] = v;
}
}
void dfs_2(int u,int tp)
{
top[u] = tp;
id[u] = ++topw;
rnk[topw]=u;
if(son[u]) dfs_2(son[u],tp);
for(int i=0;i<G[u].size();i++)
{
int v = G[u][i];
if(v == fa[u]|| v==son[u]) continue;
dfs_2(v,v);
}
}
int add[maxn<<2];
void push_down(int rt,int l,int r)
{
if(add[rt])
{
int mid=(l+r)>>1;
sum[rt<<1]+=(mid-l+1)*add[rt];
sum[rt<<1|1]+=(r-mid)*add[rt];
add[rt<<1] += add[rt];
add[rt<<1|1] += add[rt];
add[rt] = 0;
}
}
void push_up(int rt)
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
add[rt] = 0;
sum[rt]=0;
if(l == r)
{
sum[rt] =a[rnk[l]];
return ;
}
int m = (l+r)>>1;
build(lson);
build(rson);
push_up(rt);
}
void update(int L,int R,int c,int l,int r,int rt)
{
if(L<=l && R>= r)
{
add[rt] += c;
sum[rt]+=(r-l+1)*c;
return ;
}
push_down(rt,l,r);
int m = (l+r)>>1;
if(L<=m) update(L,R,c,lson);
if(R> m) update(L,R,c,rson);
push_up(rt);
}
int query(int x,int l,int r,int rt)
{
if(l == r)
{
return sum[rt];
}
push_down(rt,l,r);
int m = (l+r)>>1;
if(x <= m) query(x,lson);
else query(x,rson);
}
void change(int u,int v,int add)
{
int fu = top[u],fv = top[v];
while(fu != fv)
{
if(dep[fu] < dep[fv])
{
swap(u,v);swap(fu,fv);
}
update(id[fu],id[u],add,1,topw,1);
u = fa[fu];
fu = top[u];
}
if(u == v)
{
update(id[u],id[v],add,1,topw,1);
return;
}
else{
if(dep[u] > dep[v]) swap(u,v);
update(id[u],id[v],add,1,topw,1);
return ;
}
}
int main()
{
while(~scanf("%d%d%d",&N,&M,&P))
{
for(int i=1;i<=N;i++) scanf("%d",&a[i]);
for(int i=1,a,b;i<=M;i++)
{
scanf("%d%d",&a,&b);
G[a].push_back(b);
G[b].push_back(a);
}
topw = 0;
dfs_1(1,0,1);
dfs_2(1,1);
build(1,topw,1);
char op[10];
int a,b,c;
for(int i=0;i<P;i++)
{
scanf("%s",op);
if(op[0] == 'I')
{
scanf("%d%d%d",&a,&b,&c);
change(a,b,c);
}
else if(op[0] == 'D')
{
scanf("%d%d%d",&a,&b,&c);
change(a,b,-c);
}
else if(op[0] == 'Q')
{
scanf("%d",&a);
int ans = query(id[a],1,topw,1);
printf("%d\n",ans);
}
}
for(int i=0;i<maxn;i++) G[i].clear();
}
}