题意:
I,从 a-》b的路线上+c
D,从a->b的路线上+c
Q, 点权和
思路:
树链剖分。第一次学。
#pragma comment(linker, "/STACK:102400000,102400000")
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
const int maxn = 50005;
int val_pre[maxn],val[maxn],siz[maxn],son[maxn],id[maxn],fa[maxn],top[maxn],dep[maxn];
int topw;
int M,N,P;
vector <int >vec[maxn];
void dfs1(int u,int f,int d) //第一次搜索建立父子关系,以及节点个数关系
{
siz[u]=1;
fa[u]=f;
dep[u]=d;
son[u]=0;
for(int i=0;i<vec[u].size();i++)
{
int v=vec[u][i];
if(v==fa[u]) 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]=++topw;
if(son[u])
dfs2(son[u],tp);
for(int i=0;i<vec[u].size();i++)
{
int v=vec[u][i];
if(v==fa[u]||v==son[u]) continue;
dfs2(v,v);
}
}
//搜索基础上,线段树
struct node
{
int left,right,sum,lazy;
}tree[maxn<<2];
void build(int i,int left,int right)
{
tree[i].left=left,tree[i].right=right;
tree[i].lazy=0;
tree[i].sum=0;
if(left==right)
{
tree[i].sum=val[left];
return ;
}
int mid=(tree[i].left+tree[i].right)>>1;
build(i<<1,left,mid);
build(i<<1|1,mid+1,right);
}
void push_down(int i)
{
if(tree[i].lazy)
{
tree[i<<1].lazy+=tree[i].lazy;
tree[i<<1|1].lazy+=tree[i].lazy;
tree[i<<1].sum+=(tree[i<<1].right-tree[i<<1].left+1)*tree[i].lazy;
tree[i<<1|1].sum+=(tree[i<<1|1].right-tree[i<<1|1].left+1)*tree[i].lazy;
tree[i].lazy=0;
}
return ;
}
void update(int i,int left,int right,int w)
{
if(tree[i].left>=left&&tree[i].right<=right)
{
tree[i].lazy+=w;
tree[i].sum+=w*(tree[i].right-tree[i].left+1);
return ;
}
if(tree[i].lazy!=0)
{
push_down(i);
}
int mid=(tree[i].left+tree[i].right)>>1;
if(left<=mid)
update(i<<1,left,right,w);
if(right>mid)
update(i<<1|1,left,right,w);
}
int query(int i ,int aim)
{
if(tree[i].left==tree[i].right&&tree[i].left==aim)
{
return tree[i].sum;
}
if(tree[i].lazy)
push_down(i);
int mid=(tree[i].left+tree[i].right)>>1;
if(aim<=mid)
{
return query(i<<1,aim);
}
else
return query(i<<1|1 ,aim);
}
void change(int u,int v,int addc)
{
int fu = top[u],fv = top[v];
//swap语句保证深度小的在左侧,大的在右侧
while(fu != fv) //直到同一条重链退出while
{
if(dep[fu] < dep[fv])
{
swap(u,v);swap(fu,fv);
}
update(1,id[fu],id[u],addc);
u = fa[fu];
fu = top[u];
}
if(dep[u] > dep[v]) swap(u,v);
update(1,id[u],id[v],addc);
return ;
}
int main()
{
while(~scanf("%d%d%d",&N,&M,&P))
{
for(int i=0;i<maxn;i++)
vec[i].clear();
for(int i=1;i<=N;i++)
scanf("%d",&val_pre[i]);
int u,v;
for(int i=1 ;i<=M;i++)
{
scanf("%d%d",&u,&v);
vec[u].push_back(v);
vec[v].push_back(u);
}
topw = 0;
dfs1(1,0,1);
dfs2(1,1);
for(int i=1;i<=N;i++)
{
val[id[i]] = val_pre[i];
}
build(1,1,topw );
//debug();
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(1,id[a]);
printf("%d\n",ans);
}
}
}
}
/*
6 5 5
1 2 3 4 5 6
2 1
2 3
1 4
4 5
2 6
D 5 6 1
*/