[bzoj 4381--POI2015]Odwiedziny

給定一棵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統計出每一次行走時經過的所有點的權值和。

這道題很套路啊，它的暴力複雜度爲$n*\frac{n}{k}*\log_2n$，這種形式就得想到分塊啊！
當k小於sqrt(n)時，可以預處理一個f數組，f[x][i]表示起點爲x，終點爲根節點，步伐爲i時行走時經過的所有點的權值和。然後就可以直接在算出lca後靠運用這個f計算答案了。
而當k大於sqrt(n)時，直接暴力向上跳，統計答案。
最後複雜度就可以通過了。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0' && ch<='9')x=x*10+ch-'0',ch=getchar();
    return x*f;
}
inline void write(int x)
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);
    putchar(x%10+'0');
}
struct node
{
    int x,y,next;
}a[100010];int len,last[50010];
inline void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
int cnt,w[50010],p[50010],dep[50010],fa[50010][18],f[50010][250],ff[50010][250];
inline int jump(int x,int d)
{
    for(int i=15;i>=0;i--)if(d>=(1<<i))x=fa[x][i],d-=(1<<i);
    return x;
}
inline void dfs(int x)
{
    for(int i=1;(1<<i)<=dep[x];i++)fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y==fa[x][0])continue;
        fa[y][0]=x;dep[y]=dep[x]+1;
        for(int i=1;i<=cnt;i++)ff[y][i]=ff[x][i-1],f[y][i]=f[ff[y][i]][i]+w[y];
        dfs(y);
    }
}
inline int LCA(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    for(int i=15;i>=0;i--)if(dep[x]-dep[y]>=(1<<i))x=fa[x][i];
    if(x==y)return x;
    for(int i=15;i>=0;i--)if(dep[x]>=(1<<i) && fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
inline int solve(int x,int y,int d)
{
    int ans=0,P=LCA(x,y),tot=dep[x]+dep[y]-2*dep[P];if(tot%d!=0)ans=w[y],y=ff[y][tot%d];
    int dx=(dep[x]-dep[P])%d,dy=(dep[y]-dep[P])%d;
    if(dx==0 || dy==0)ans+=w[P];
    if(d<=cnt)ans+=f[x][d]-f[ff[P][dx]][d]+f[y][d]-f[ff[P][dy]][d];
    else
    {
        while(dep[x]>dep[P])ans+=w[x],x=jump(x,d);
        while(dep[y]>dep[P])ans+=w[y],y=jump(y,d);
    }
    return ans;
}
int main()
{
    //freopen("a.in","r",stdin);
    //freopen("a.out","w",stdout);
    int n=read();cnt=sqrt(n);
    for(int i=1;i<=n;i++)w[i]=read(),ff[i][0]=i;
    for(int i=1;i<n;i++)
    {
        int x=read(),y=read();
        ins(x,y),ins(y,x);
    }dfs(1);
    for(int i=1;i<=n;i++)p[i]=read();
    for(int i=1;i<n;i++)
    {
        int d=read();
        printf("%d\n",solve(p[i],p[i+1],d));
    }
    return 0;
}