[Luogu 5465] [LOJ 6435] [PKUSC2018]星际穿越(倍增)
题面
n个点的图,点i和[l[i],i)的所有点连双向边。每次询问(l,r,x)表示x到[l,r]的所有点的最短路径长度和/(r-l+1)。
(n leq 10^5,l_i<r_i<x_i)
分析
有(du)趣(liu)的倍增问题。
观察到(l_i<r_i<x_i),也就是说我们要求往左走的最小距离。首先差分一下,设(sum(x,i))表示(x)到([i,x-1])的最短距离和。那么答案就是(frac{sum(x,l)-sum(x,r+1)}{r-l+1})
然后我们不要直接考虑到x的最短路,而是反过来考虑走k步可以到达哪些节点。
我们从x开始走1步,可以走到哪里呢?。可以跳到的编号最小的点为(l_x),而最大的点应该是(rb(x)=max(k) (l_k leq x)),因为只要(l_k leq x), (x)与(k)之间就有一条边。因此第1步能到达的区间是([l_x,rb(x)])
第2步呢?。根据上面的分析,应该是(min(l_i)(i in [l_x,rb(x)])).但是实际上可以简化为(min(l_i)(i in [l_x,n])).因为在([rb(x),n])内的点,它们的(l)比x还大,自然也比([l_x,rb(x)])内的点的(l)还大,对最小值没有任何影响。
因此,设走(k(k>1))步可以到达的编号最小的点为(a),那么走(k+1)步能够到达的编号最小的点为(min(l_i)(iin [a,n])).这样就可以倍增优化。
设(f[i][j])表示([i,n])内节点走(2^j)步能够到达的编号最小的点。那么显然有:
为了求答案,我们再维护一个和,(g[i][j])表示(i)到([f[i][j],i-1])内所有点的距离和。那么:
(g[i][0]=i-f[i][0]) (到([f[i][0],i-1])中的每个点的距离都为1)
这是因为:到([f[i][j-1],i-1])内的节点,距离和是(g[i][j-1])。而到([f[i][j],f[i][j-1]])内的节点的距离由两部分组成,一部分是各节点到(f[i][j-1])的距离(g[f[i][j-1]][j-1]),另一部分是从(f[i][j-1])到(i)的距离(2^{j-1}).因为一共有((f[i][j-1]-f[i][j]))个节点,所以要乘上((f[i][j-1]-f[i][j]))
查询的话就倍增的跳即可,注意一些细节,还是直接上代码吧.
int calc(int x,int l){ //sum(x,l),[l,x-1]内答案
if(a[x]<=l) return x-l;//只需走一次的情况,特判
//第一次走的答案
int ans=x-a[x];//距离和
int cnt=1;//走的次数
x=a[x];
for(int i=log2n;i>=0;i--){
if(f[x][i]>=l){
ans+=g[x][i]+cnt*(x-f[x][i]);//,g[x][i]为当前这段的距离和,但是之前每个节点还跳了cnt步到x,因此要加上cnt*[f[x][i],x-1]
cnt+=(1<<i);
x=f[x][i];
}
}
if(x>l) ans+=x-l+cnt*(x-l);//如果最后一步没跳满,加上f[x][i]
return ans;
}
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define maxn 300000
#define maxlogn 25
using namespace std;
int gcd(int a,int b){
return b==0?a:gcd(b,a%b);
}
int n,q;
int log2n;
int a[maxn+5];
int f[maxn+5][maxlogn+5];
int g[maxn+5][maxlogn+5];
int calc(int x,int l){ //[l,x-1]内答案
if(a[x]<=l) return x-l;
//特判第一次跳
int ans=x-a[x];
int cnt=1;
x=a[x];
for(int i=log2n;i>=0;i--){
if(f[x][i]>=l){
ans+=g[x][i]+cnt*(x-f[x][i]);//[f[x][i],x-1]跳到x还需cnt步
cnt+=(1<<i);
x=f[x][i];
}
}
if(x>l) ans+=x-l+cnt*(x-l);//如果最后一步没跳满,加上f[x][i]
return ans;
}
int main(){
int l,r,x;
scanf("%d",&n);
log2n=log2(n)+1;
a[1]=1;
for(int i=2;i<=n;i++) scanf("%d",&a[i]);
f[n][0]=a[n];
for(int i=n-1;i>=1;i--){
f[i][0]=min(f[i+1][0],a[i]);
g[i][0]=i-f[i][0];
}
for(int j=1;j<=log2n;j++){
for(int i=1;i<=n;i++){
if(f[i][j-1]){
f[i][j]=f[f[i][j-1]][j-1];
g[i][j]=g[i][j-1]+g[f[i][j-1]][j-1]+((f[i][j-1]-f[i][j])<<(j-1));
//[f[i][j],f[i][j-1]-1]内的点跳到j还需2^{j-1}步
}
}
}
scanf("%d",&q);
while(q--){
scanf("%d %d %d",&l,&r,&x);
int up=calc(x,l)-calc(x,r+1);
int down=r-l+1;
int g=gcd(up,down);
up/=g;
down/=g;
printf("%d/%d
",up,down);
}
}