弹飞绵羊 分块

题目描述

某天，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的情况，你都要输出一个需要的步数，占一行。

做法：进行分块，f[i]表示从i开始弹，弹几次可以弹出当前的块。to[i]表示弹出当前块时到达下一个块的位置在哪里。

在查询时我们进行整块的模拟，把f值加一起即可。

在修改时，我们只需要修改l[belong[x]]到x之间的f和to值就行了。

代码如下：

#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdio>
#define N 200005
#define M 100005
using namespace std;
int n,Q,num,block,op,x,y;
int k[N],r[N],l[N],belong[N],f[N],to[N];
void change(int x,int y)
{
	k[x]=y;
	for (int i=x;i>=l[belong[x]];i--){
		if (i+k[i]>=l[belong[i]+1]) f[i]=1,to[i]=i+k[i];
		else f[i]=f[i+k[i]]+1,to[i]=to[i+k[i]];
	}
    return;
}
int sum(int x)
{
	int now=x,ans=0;
	while (now<=n){
		ans=ans+f[now];
		now=to[now];
	}
	return ans;
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++) scanf("%d",&k[i]);
	block=(int)sqrt(n);
	num=n/block;
	if (n%block!=0) num++;
	for (int i=1;i<num;i++){
		l[i]=(i-1)*block+1;
		r[i]=i*block;
	}
	r[num]=n; l[num]=r[num-1]+1;
	for (int i=1;i<=n;i++) belong[i]=((i-1)/block)+1;
	for (int i=1;i<=n-1;i++){
		int now=belong[i];
		int x=i;
		while (x<=r[now]){
			f[i]++;
			x=x+k[x];
			to[i]=x;
		}
	}
	f[n]=1; to[n]=n+1;
	scanf("%d",&Q);
	while (Q--){
		scanf("%d",&op);
		if (op==1){
			scanf("%d",&x);
			printf("%d\n",sum(x+1));
		}
		else {
			scanf("%d%d",&x,&y);
			change(x+1,y);
		}
	}
	return 0;
}