【BZOJ】4034 [HAOI2015]树上操作 树链剖分

题目传送门

这题的题目意思很明确，就是树链剖分。

但是，这题需要用到long long，在打lazy标记时需要强转，把int转成long long，我也在这里错了不下15次。

博主花了2天来改这道题，并和许多大佬的标程对比，并没有发现任何错误。偶然中改了这个地方，发现程序AC后，当即掀桌子……

博主对此表示很难受……

附上AC代码：

#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define N 100010
#define lt (k<<1)
#define rt (k<<1|1)
#define mid ((l+r)>>1)
using namespace std;

struct side{
	int to,nt;
}s[N*2];
struct tree{
	long long sum,lz;
}t[N*4];
int n,m,a[N],num,h[N],x,y,size[N],f[N],dis[N],hs[N],wz[N],top[N],id,o,w;

void read(int& a){
	static char c=getchar();a=0;int f=1;
	while (!isdigit(c)) {if (c=='-') f=-1;c=getchar();}
	while (isdigit(c)) a=a*10+c-'0',c=getchar();
	a*=f;return;
}

void add(int x,int y){
	s[num]=(side){y,h[x]},h[x]=num++;
	s[num]=(side){x,h[y]},h[y]=num++;
}

void so(int x){
	size[x]=1,dis[x]=dis[f[x]]+1;
	for (int i=h[x]; ~i; i=s[i].nt)
		if (s[i].to!=f[x]){
			f[s[i].to]=x,so(s[i].to),size[x]+=size[s[i].to];
			if (size[s[i].to]>size[hs[x]]) hs[x]=s[i].to;
		}
	return;
}

void so(int x,int fa){
	top[x]=fa,wz[x]=++id;
	if (!hs[x]) return;
	so(hs[x],fa);
	for (int i=h[x]; ~i; i=s[i].nt)
		if (s[i].to!=f[x]&&s[i].to!=hs[x])
			so(s[i].to,s[i].to);
	return;
}

void push(int k,int l,int r){
	t[lt].lz+=t[k].lz,t[lt].sum+=t[k].lz*(mid-l+1);
	t[rt].lz+=t[k].lz,t[rt].sum+=t[k].lz*(r-mid);
	t[k].lz=0;return;
}

void change(int k,int l,int r,int ql,int qr,int w){
	if (l>qr||r<ql) return;
	if (l>=ql&&r<=qr){
		t[k].sum+=(long long)(r-l+1)*w,t[k].lz+=w;
		return;
	}
	push(k,l,r);
	change(lt,l,mid,ql,qr,w),change(rt,mid+1,r,ql,qr,w);
	t[k].sum=t[lt].sum+t[rt].sum;
	return;
}

long long query(int k,int l,int r,int ql,int qr){
	if (l>qr||r<ql) return 0;
	if (l>=ql&&r<=qr) return t[k].sum;
	push(k,l,r);
	return query(lt,l,mid,ql,qr)+query(rt,mid+1,r,ql,qr);
}

long long find(int x,int y){
	long long sum=0;
	for (int fx=top[x],fy=top[y]; fx!=fy; x=f[top[x]],fx=top[x]){
		if (dis[fx]<dis[fy]) swap(fx,fy),swap(x,y);
		sum+=query(1,1,n,wz[fx],wz[x]);
	}
	if (dis[x]>dis[y]) swap(x,y);
	return sum+query(1,1,n,wz[x],wz[y]);
}

int main(void){
	read(n),read(m),memset(h,-1,sizeof h);
	for (int i=1; i<=n; ++i) read(a[i]);
	for (int i=1; i<n; ++i) read(x),read(y),add(x,y);
	so(1),so(1,1);
	for (int i=1; i<=n; ++i) change(1,1,n,wz[i],wz[i],a[i]);
	while (m--){
		read(o),read(x);
		switch (o){
			case 1:{
				read(w),change(1,1,n,wz[x],wz[x],w);
				break;
			}
			case 2:{
				read(w),change(1,1,n,wz[x],wz[x]+size[x]-1,w);
				break;
			}
			case 3:{
				printf("%lld\n",find(x,1));
				break;
			}
		}
	}
	return 0;
}