bzoj4381 [POI2015]Odwiedziny 分块+倍增

Description

---

给定一棵n个点的树，树上每条边的长度都为1，第i个点的权值为a[i]。
Byteasar想要走遍这整棵树，他会按照某个1到n的全排列b走n-1次，第i次他会从b[i]点走到b[i+1]点，并且这一次的步伐大小为c[i]。
对于一次行走，假设起点为x，终点为y，步伐为k，那么Byteasar会从x开始，每步往前走k步，如果最后不足k步就能到达y，那么他会一步走到y。
请帮助Byteasar统计出每一次行走时经过的所有点的权值和。

Solution

---

由于不会什么高级做法，我们考虑分块

对于$\ge \sqrt{n}$的c我们直接走，否则预处理s[x][k]表示x向上步伐为k走到根的答案，减一减就可以了
向上爬k步可以倍增也可以长链剖分，然鹅直接倍增就能过。。

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#define rep(i,st,ed) for (register int i=st;i<=ed;++i)
#define drp(i,st,ed) for (register int i=st;i>=ed;--i)

const int N=50005;

struct edge {int y,next;} e[N*2];

int s[N][235],w[N],b[N],c[N];
int dep[N],fa[N][16];
int ls[N],edCnt,B;

int read() {
	int x=0,v=1; char ch=getchar();
	for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
	for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
	return x*v;
}

void add_edge(int x,int y) {
	e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
	e[++edCnt]=(edge) {x,ls[y]}; ls[y]=edCnt;
}

void dfs1(int x) {
	rep(i,1,15) fa[x][i]=fa[fa[x][i-1]][i-1];
	for (int i=ls[x];i;i=e[i].next) {
		if (e[i].y==fa[x][0]) continue;
		fa[e[i].y][0]=x; dep[e[i].y]=dep[x]+1;
		dfs1(e[i].y);
	}
}

int get_lca(int x,int y) {
	if (dep[x]<dep[y]) std:: swap(x,y);
	drp(i,15,0) if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
	if (x==y) return x;
	drp(i,15,0) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}

int get_up(int x,int k) {
	drp(i,15,0) if ((k>>i)&1) x=fa[x][i];
	return x;
}

void dfs2(int x) {
	rep(i,1,B) {
		int up=get_up(x,i);
		s[x][i]+=s[up][i]+w[x];
	}
	for (int i=ls[x];i;i=e[i].next) {
		if (e[i].y==fa[x][0]) continue;
		dfs2(e[i].y);
	}
}

int ask(int x,int y,int k) {
	if (x==y) return w[x];
	int lca=get_lca(x,y),res=0;
	if (k>B) {
		for (int nw=x;dep[nw]>=dep[lca];) {
			res+=w[nw];
			nw=get_up(nw,k);
		}
		int t=get_up(y,(dep[x]+dep[y]-dep[lca]*2)%k);
		for (int nw=t;dep[nw]>=dep[lca];) {
			res+=w[nw];
			nw=get_up(nw,k);
		}
		if ((dep[x]+dep[y]-dep[lca]*2)%k) res+=w[y];
		if ((dep[x]-dep[lca])%k==0) res-=w[lca];
	} else {
		res+=s[x][k]-s[get_up(lca,k-(dep[x]-dep[lca])%k)][k];
		int t=get_up(y,(dep[x]+dep[y]-dep[lca]*2)%k);
		if (dep[t]>=dep[lca]) res+=s[t][k]-s[get_up(lca,k-(dep[t]-dep[lca])%k)][k];
		if ((dep[x]+dep[y]-dep[lca]*2)%k) res+=w[y];
		if ((dep[x]-dep[lca])%k==0) res-=w[lca];
	}
	return res;
}

int main(void) {
	freopen("data.in","r",stdin);
	freopen("myp.out","w",stdout);
	int n=read(),ans=0; B=sqrt(n);
	rep(i,1,n) w[i]=read();
	rep(i,2,n) add_edge(read(),read());
	dfs1(dep[1]=1); dfs2(1);
	rep(i,1,n) b[i]=read();
	rep(i,1,n-1) c[i]=read();
	rep(i,1,n-1) printf("%d\n", ask(b[i],b[i+1],c[i]));
	return 0;
}