Odwiedziny[POI 2015]

题目描述

给定一棵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统计出每一次行走时经过的所有点的权值和。

输入

• Line 1：一个正整数n(2<=n<=50000)。表示节点的个数。
• Line 2：n个正整数，其中第i个数为ai(1≤ai≤10000)，分别表示每个点的权值。
• Line 3~n+1：包含两个正整数u,v(1≤u,v≤n)，表示u与v之间有一条边。
• Line n+2：n个互不相同的正整数，其中第i个数为b[i](1≤b[i]≤n)，表示行走路线。
• Line n+3：n-1个正整数，其中第i个数为ci(1≤ci<n)，表示每次行走的步伐大小。

 

输出

包含n-1行，每行一个正整数，依次输出每次行走时经过的所有点的权值和

SOL：

我们观察了一波局势，发现这道题有些眼熟。相比大家应该都做过这样一道题：给你一个静态的序列（所谓静态是指这个区间不会做任何的修改），M次给出X,L,K，让你求从X起的L个数的和，这L个数中间两两下表相间K（举个栗子：X=5，L=4，K=3，那么sum=a5+a8+a11+a14）。（N<=10W，M<=10W）

这道题我们的思想就是分块，我们发现当K很大时，我们暴力求和的速度是相当快的。因为我们发现 L<=N/K，我们的暴力复杂度是O(L)的。而K很小时，暴力就很慢了，接近于O(N)。我们又注意到K=1时便是前缀和，可以做到O（1）查询。那么我们不禁想当K很小时我们用前缀和优化（K>1有间距的前缀和相比大家都会）。

我们假设我们采取这样的策略，当K<S时我们用前缀和，K>S时我们采取暴力，那么我们就可以在最坏情况下O（n*s+m*n/s）下解决问题,那么我们解得S=sqrt N时最优。（实际上S取的稍微小一些会更快，因为我们刚刚计算的是最坏情况）

那么我们就可以做这一道题了，同样的思想，当K<sqrt N 时我们预处理，K > sqrt N 时我们便暴力向上跳。

 case 1: K<sqrt N 我们预处理每个点 以K为间距向上跳的 答案，x的答案加上y的答案减去lca的答案就是了。

case 1: K>sqrt N 我们暴力做，我们每次用树上倍增的方法 log N 的方法向上跳，暴力统计答案，跳跳就好了。

看代码：

#include<bits/stdc++.h>
#define sight(c) ('0'<=c&&c<='9')
#define eho(x) for (int i=head[x];i;i=net[i])
#define N 50007
#define BRE 237
#define SIZ 21
int a[N],fall[N<<1],net[N<<1],head[N],tot,f[SIZ][N],dep[N],val[N][BRE],g[BRE][N],
m,rog,ot,r,ans,n,x,y,len[N],k;
inline void read(int &x){
    static char c;
    for (c=getchar();!sight(c);c=getchar());
    for (x=0;sight(c);c=getchar()) x=x*10+c-48;
}
using namespace std;
inline void add(int x,int y){fall[++tot]=y; net[tot]=head[x]; head[x]=tot;}
void dfs(int x,int fa){
   f[0][x]=fa;dep[x]=dep[fa]+1;
    eho(x) if(fall[i]^fa)  dfs(fall[i],x);
}
void Dfs(int x){
   for (int i=1;i<=m;i++) val[x][i]=val[g[i][x]][i]+a[x];
   eho(x) if (fall[i]^f[0][x]) Dfs(fall[i]);
}
void write(int x){if (x<10) {putchar('0'+x); return;} write(x/10); putchar('0'+x%10);}
inline void writeln(int x){ if (x<0) putchar('-'),x*=-1; write(x); putchar('\n'); }
inline int lca(int x,int y){
    if (dep[x]<dep[y]) swap(x,y);
    for (int i=SIZ-1;~i;i--) if (dep[f[i][x]]>=dep[y]) x=f[i][x];
    if (x==y) return x;
    for (int i=SIZ-1;~i;i--) if (f[i][x]^f[i][y]) x=f[i][x],y=f[i][y];
    return f[0][x];
}
inline int up(int x,int k) {if (k<=m)return g[k][x];for (int i=SIZ-1;~i;i--) if ((k>>i)&1) x=f[i][x];return x;}
inline int rop(int x,int y,int k) {
    if (dep[x]<dep[y]) return 0;rog=0;
    if (k<m) {
        rog+=val[x][k]; ot=k-(dep[x]-dep[y])%k; rog-=val[k==ot?y:up(y,ot%k)][k];
    } else while (dep[x]>dep[y]) rog+=a[x],x=up(x,k);
    return rog;
}
void solve(int x,int y,int k){
    int L=lca(x,y); r=(dep[x]+dep[y]-(dep[L]<<1))%k;
    ans=rop(x,L,k);
    if (r) ans+=a[y],y=up(y,k);
    ans+=rop(y,f[0][L],k);
    writeln(ans);
}
int main () {
    freopen("a.in","r",stdin);
    read(n); m=min((int)sqrt(n),BRE-2);
    for (int i=1;i<=n;i++) read(a[i]),g[0][i]=a[i];
    for (int i=n-1;i;i--) read(x),read(y),add(x,y),add(y,x);
    dfs(1,0);
    for (int j=1;j<SIZ;j++)
     for (int i=1;i<=n;i++) f[j][i]=f[j-1][f[j-1][i]];
    memcpy(g[1],f[0],sizeof f[0]);
    for (int j=2;j<=m;j++)
     for (int i=1;i<=n;i++) g[j][i]=g[1][g[j-1][i]];
    Dfs(1);
    for (int i=1;i<=n;i++) read(len[i]);
    for (int i=1;i<n;i++) read(k),
     solve(len[i],len[i+1],k);
    return 0;
}