Odwiedziny[POI 2015]

Posted 殇雪

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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(1ai10000),分别表示每个点的权值。
  • Line 3~n+1:包含两个正整数u,v(1u,vn),表示u与v之间有一条边。
  • Line n+2:n个互不相同的正整数,其中第i个数为b[i](1b[i]n),表示行走路线。
  • Line n+3:n-1个正整数,其中第i个数为ci(1ci<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;
}

 

 

 

以上是关于Odwiedziny[POI 2015]的主要内容,如果未能解决你的问题,请参考以下文章

@bzoj - 4381@ [POI2015] Odwiedziny

Odwiedziny[POI 2015]

[POI2015]LOG

[POI2015]PUS

java poi xwpf操作word生成一个表格怎么合并单元格,求大神指导!

VS2015 代码片段整理