题面:
题目描述
某天,Lostmonkey发明了一种超级弹力装置,为了在他的绵羊朋友面前显摆,他邀请小绵羊一起玩个游戏。游戏一开始,Lostmonkey在地上沿着一条直线摆上n个装置,每个装置设定初始弹力系数ki,当绵羊达到第i个装置时,它会往后弹ki步,达到第i+ki个装置,若不存在第i+ki个装置,则绵羊被弹飞。绵羊想知道当它从第i个装置起步时,被弹几次后会被弹飞。为了使得游戏更有趣,Lostmonkey可以修改某个弹力装置的弹力系数,任何时候弹力系数均为正整数。
输入输出格式
输入格式:
第一行包含一个整数n,表示地上有n个装置,装置的编号从0到n-1,接下来一行有n个正整数,依次为那n个装置的初始弹力系数。第三行有一个正整数m,接下来m行每行至少有两个数i、j,若i=1,你要输出从j出发被弹几次后被弹飞,若i=2则还会再输入一个正整数k,表示第j个弹力装置的系数被修改成k。对于20%的数据n,m<=10000,对于100%的数据n<=200000,m<=100000
输出格式:
对于每个i=1的情况,你都要输出一个需要的步数,占一行。
输入输出样例
输入样例#1:
4
1 2 1 1
3
1 1
2 1 1
1 1
输出样例#1:
2
3
题解
这题的正解是lct,然而并不会,所以只好用一些猥琐的方法来解题。
于是就找到了分块的题解,学习了一下。
因为这题的n不算很大,所以
nn−−√
的复杂度还是可以接受的,于是我们就可以用分块来解决。我们可以记录一个sum[i]表示第i个点要弹多少次才能弹出这个块,因为每个块的大小是
n−−√
,所以每次修正的复杂度就是
n−−√
,不会超时了。每次查询就是简单累加每个块弹出的步数就好了。
代码如下:
#include<cstdio>
#include<string>
#include<cmath>
using namespace std;
const int maxn=200005,maxm=505;
int n,m,Q,num[maxn],lnk[maxn],sum[maxn],outt[maxn];
struct dyt{int l,r;}a[maxm];
inline int read(){
int x=0; char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
return x;
}
void doit(int l,int r){
for (int i=r;i>=l;i--)
if (i+num[i]>a[lnk[i]].r) sum[i]=1,outt[i]=i+num[i];
else sum[i]=sum[i+num[i]]+1,outt[i]=outt[i+num[i]];
}
int main(){
//freopen("2002.in","r",stdin);
n=read(); m=sqrt(n); if (m*m<n) m++;
int s=0;
for (int i=1;i<=n;i++) num[i]=read();
for (int i=1;i<=n;i+=m) a[++s].l=i,a[s].r=i+m-1;
if (s<m) a[++s].l=s*m+1,a[s].r=n; s=1;
for (int i=1;i<=n;i++) {if (i>a[s].r) s++; lnk[i]=s;}
doit(1,n);
Q=read();
while (Q--) {
int x=read(),y=read()+1;
if (x==1) {
int ans=sum[y],x=outt[y];
for (int i=lnk[y];i<=m&&x<=n;i++) ans+=sum[x],x=outt[x];
printf("%d\n",ans);
} else {
int z=read(); num[y]=z; doit(a[lnk[y]].l,a[lnk[y]].r);
}
}
return 0;
}