【BZOJ】[HAOI2015]树上操作-DFS序

传送门:点击打开链接

 

题意:

有一棵点数为 N 的树,以点 1 为根,且树点有边权。有 M 个操作,分为三种:

操作 1 :把某个节点 x 的点权增加 a 。
操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a 。

操作 3 :询问某个节点 x 到根的路径中所有点的点权和。

数据范围:

 N,M<=100000 ,且所有输入数据的绝对值都不会超过 10^6 。

DFS序的板子题,网上博客很多,都比较好理解。

代码实现的时候用线段树来二分排dfs序。要注意的是,dfs序in和out要同时更新,注意正负。

用的是朴素做法,据说可以树链剖分,可我不会啊。才学就打一个朴素版吧。

//3.11:我来补代码了:树链剖分:点击打开链接

代码

#include
using namespace std;
typedef long long ll;
const int N=1e5+2;
int n,m,root=1,cnt,tot,ini[N],num;
int in[N],out[N],df[600010];
int to[N<<1],next[N<<1],head[N<<1];

struct node{
	int l,r;
	int fu,zh;//zh->zheng_fu->fu
	ll sum,add;
}t[2400010];

inline void link(int u,int v){
	++tot;
	next[tot]=head[u];
	head[u]=tot;
	to[tot]=v;
}

inline void pushdown(int x){
  if(t[x].add==0) return;
    t[x<<1].sum+=1ll*(t[x<<1].zh-t[x<<1].fu)*t[x].add;
    t[x<<1|1].sum+=1ll*(t[x<<1|1].zh-t[x<<1|1].fu)*t[x].add;
   	t[x<<1].add+=t[x].add;
   	t[x<<1|1].add+=t[x].add;
   	t[x].add=0;
}

inline void maintain(int x){
	t[x].sum=t[x<<1].sum+t[x<<1|1].sum;
}

inline void dfs(int fa,int k)
{
	df[++cnt]=k;in[k]=cnt;
    for(int i=head[k];i;i=next[i]){
		if(to[i]!=fa)
		  dfs(k,to[i]);
    }
    df[++cnt]=-k;out[k]=cnt;
}

inline void build(int k,int l,int r){
	t[k].l=l,t[k].r=r;t[k].sum=t[k].add=0;
	if(l==r){
		if(df[l]>0)
		  t[k].sum=(ll)ini[df[l]],t[k].zh++;
		else t[k].sum=-(ll)ini[-df[l]],t[k].fu++;
	}else{
		int mid=(l+r)>>1;
		build(k<<1,l,mid);
		build(k<<1|1,mid+1,r);
		maintain(k);
		t[k].zh=t[k<<1].zh+t[k<<1|1].zh;
		t[k].fu=t[k<<1].fu+t[k<<1|1].fu;
	}
}

inline void ad(int k,int L,int R,ll addv)
{
	if(t[k].l>=L && t[k].r<=R){
		t[k].sum+=1ll*(t[k].zh-t[k].fu)*addv;
	    t[k].add+=addv;
	}else{
		pushdown(k);
		int mid=(t[k].l+t[k].r)>>1;
	    if(R<=mid) ad(k<<1,L,R,addv);
	    else if(L>mid) ad(k<<1|1,L,R,addv);
	    else {
		  ad(k<<1,L,mid,addv);
		  ad(k<<1|1,mid+1,R,addv);
		}
	    maintain(k);
	}
}

inline ll query(int k,int L,int R)
{
	if(t[k].l>=L && t[k].r<=R)
		return t[k].sum;
	int mid=(t[k].l+t[k].r)>>1;pushdown(k);
	if(R<=mid) return query(k<<1,L,R);
	else if(L>mid)
		return query(k<<1|1,L,R);
	else return 
		 query(k<<1,L,mid) + query(k<<1|1,mid+1,R);
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	  scanf("%d",&ini[i]);
	for(int x,y,i=1;i

 

你可能感兴趣的:(线段树)