今天是2017/5/22,DCDCBigBig的第九篇博文
树链剖分(+线段树)
#include
#include
#include
#include
#include
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
struct edge{
int v,next;
}a[2000001];
int n,m,q,u,v,ww,now=0,tot=0,sum[2000001],cal[2000001],num[200001],head[200001],son[200001],siz[200001],dep[200001],fa[200001],top[200001],vis[200001],w[200001];
char ord[10];
int max(int a,int b){
return a>b?a:b;
}
void add(int u,int v){
a[++tot].v=v;
a[tot].next=head[u];
head[u]=tot;
}
void dfs1(int u,int f,int dp){
int v;
dep[u]=dp;
fa[u]=f;
siz[u]=1;
for(int tmp=head[u];tmp!=-1;tmp=a[tmp].next){
v=a[tmp].v;
if(v!=f){
dfs1(v,u,dp+1);
siz[u]+=siz[v];
if(son[u]==-1||siz[v]>siz[son[u]]){
son[u]=v;
}
}
}
}
void dfs2(int u,int tp){
int v;
top[u]=tp;
vis[u]=++now;
w[vis[u]]=u;
if(son[u]==-1)return;
dfs2(son[u],tp);
for(int tmp=head[u];tmp!=-1;tmp=a[tmp].next){
v=a[tmp].v;
if(v!=son[u]&&v!=fa[u])dfs2(v,v);
}
}
void pushD(int rt,int len){
if(cal[rt]){
cal[rt<<1]+=cal[rt];
cal[rt<<1|1]+=cal[rt];
sum[rt<<1]+=(len-len/2)*cal[rt];
sum[rt<<1|1]+=(len/2)*cal[rt];
cal[rt]=0;
}
}
void build(int l,int r,int rt){
cal[rt]=0;
if(l==r){
sum[rt]=num[w[l]];
return;
}
int mid=(l+r)/2;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
void updata(int rt,int k,int l,int r,int L,int R){
if(L<=l&&r<=R){
cal[rt]+=k;
sum[rt]+=k*(r-l+1);
return;
}
int mid=(l+r)/2;
pushD(rt,r-l+1);
if(L<=mid)updata(rt<<1,k,l,mid,L,R);
if(R>mid)updata(rt<<1|1,k,mid+1,r,L,R);
sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
int query(int rt,int l,int r,int aa){
if(l==r){
return sum[rt];
}
int ans=0,mid=(l+r)/2;
pushD(rt,r-l+1);
if(aa<=mid)ans=query(rt<<1,l,mid,aa);
else ans=query(rt<<1|1,mid+1,r,aa);
sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
return ans;
}
void update(int u,int v,int w){
while(top[u]!=top[v]){
if(dep[top[u]]1,w,1,n,vis[top[u]],vis[u]);
u=fa[top[u]];
}
if(dep[u]>dep[v])swap(u,v);
updata(1,w,1,n,vis[u],vis[v]);
}
int main(){
while(scanf("%d%d%d",&n,&m,&q)!=EOF){
memset(head,255,sizeof(head));
memset(son,255,sizeof(son));
tot=now=0;
for(int i=1;i<=n;i++){
scanf("%d",&num[i]);
}
for(int i=1;i<=m;i++){
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
dfs1(1,0,0);
dfs2(1,1);
build(1,n,1);
for(int i=1;i<=q;i++){
scanf("%s",ord);
if(ord[0]=='Q'){
scanf("%d",&u);
printf("%d\n",query(1,1,n,vis[u]));
}else{
scanf("%d%d%d",&u,&v,&ww);
if(ord[0]=='D')ww=-ww;
update(u,v,ww);
}
}
}
return 0;
}
因