bzoj 2002[Hnoi2010]Bounce 弹飞绵羊(分治分块)

Description

某天，Lostmonkey发明了一种超级弹力装置，为了在他的绵羊朋友面前显摆，他邀请小绵羊一起玩个游戏。游戏一开始，Lostmonkey在地上沿着一条直线摆上n个装置，每个装置设定初始弹力系数ki，当绵羊达到第i个装置时，它会往后弹ki步，达到第i+ki个装置，若不存在第i+ki个装置，则绵羊被弹飞。绵羊想知道当它从第i个装置起步时，被弹几次后会被弹飞。为了使得游戏更有趣，Lostmonkey可以修改某个弹力装置的弹力系数，任何时候弹力系数均为正整数。

Input

第一行包含一个整数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

Output

对于每个i=1的情况，你都要输出一个需要的步数，占一行。

Sample Input

4
1 2 1 1
3
1 1
2 1 1
1 1

Sample Output

2
3

 

这个题给了10s,但是也不能乱搞.

我们现将弹簧进行分块,对于每个弹簧我们记录它跳出自己所在块的步数times跟跳出当前块的落点pos

这样我们对于每次更改就可以只更改x所在块中位于x之前的点,跟x自己,这样是每个询问是

然而对于每一个询问由于我们记录的pos是跳出当前块的落点,这样最多我们跳块,所以也是

 

综合复杂度

写完这个题感觉分块真是优美

#include <bits/stdc++.h>

using namespace std;
const int maxn = 220000;
int n,m;
int q,op;
int a[maxn];
int pos[maxn],times[maxn];
int main()
{
    //freopen("de.txt","r",stdin);
    while (~scanf("%d",&n)){
        memset(pos,-1,sizeof pos);
        for (int i=0;i<n;++i){
            scanf("%d",&a[i]);
        }
        int sz = (int) sqrt(n);
        for (int i=n-1;i>=0;--i){//从后向前初始化pos,times
            int tmp = a[i]+i;
            if (tmp>n)
                pos[i]=-1,times[i]=1;//直接跳出边界
            else if (tmp>=(i/sz+1)*sz)
                pos[i] = tmp,times[i]=1;//跳出了当前块
            else
                pos[i] = pos[tmp],times[i]=times[tmp]+1;//跳到了块内的另一个点
        }
        scanf("%d",&m);
        for (int i=0;i<m;++i){
            scanf("%d",&q);
            if (q==1){
                int x;
                int ans = 0;
                scanf("%d",&x);
                for (int j=x;j!=-1;j=pos[j]){
                    ans+=times[j];
                }
                printf("%d\n",ans);
            }
            else{
                int x,y;
                scanf("%d%d",&x,&y);
                a[x]=y;
                for (int j=x;j>=x/sz*sz;j--){//更新只更新x所在块里面位于x之前的块跟x自己
                    int tmp=j+a[j];
                    if (tmp>=n)
                        pos[j]=-1,times[j]=1;
                    else if (tmp>=(j/sz+1)*sz)
                        pos[j]=tmp,times[j]=1;
                    else
                        pos[j]=pos[tmp],times[j] = times[tmp]+1;
                }
            }
        }
    }
    return 0;
}