在线oj提交传送门
第一次打分块的代码,qwq原来也不是特别麻烦
每一个点记录跳出所在块的步数,和跳出这个块之后落到哪一个点,每次修改,都把整个块的内容全部修改一遍,复杂度O(nlogn)
注意:
1.修改时一定要暴力把整个块内所有受到修改影响的点修改一遍
2.本题中记录了一个last数组表示每一个块的最后一个元素。修改时,要从该点一直到上一个块的最后一个元素的下一个元素也就是本块的第一个元素全部修改。但是如果遇到修改第一个块内的元素,由于一开始没有记录第0个块的最后一个,所以last[0]==0,但是标号0的实际上是第一个元素,这就出现没有处理到0的情况,导致WA了四组。解决办法呢就是每次修改块内信息时,连同last[num[x]]一块处理一遍(反正不会影响答案),这样就会把0处理到了。或者是直接设last[0]=-1,其他的都是处理该块内的信息就可以了
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=200000+10;
int n,tot=1,m;
int k[maxn],s[maxn],nxt[maxn],last[maxn],num[maxn];
void done(int l,int r)
{
for(int i=r;i>=l;--i)
if(i+k[i]>last[num[i]]) nxt[i]=i+k[i],s[i]=1;
else s[i]=s[i+k[i]]+1,nxt[i]=nxt[i+k[i]];
}
int ask(int x)
{
if(x>=n-1) return 1;
int i=x,ans=0;
while(i<n) ans+=s[i],i=nxt[i];
return ans;
}
void init()
{
int Size=sqrt(n);
int j=0;
for(int i=0;i<n;++i)
{
if(j==Size) last[tot]=i-1,tot++,j=1;
else j++;
num[i]=tot;
}
last[tot]=n-1;
done(0,n-1);
}
int main()
{
scanf("%d",&n);
for(int i=0;i<n;++i) scanf("%d",&k[i]);
init();
scanf("%d",&m);
for(int i=1;i<=m;++i)
{
int x,y;
scanf("%d%d",&x,&y);
if(x==1) printf("%d\n",ask(y));
else
{
int z;
scanf("%d",&z);
k[y]=z;
done(last[num[y]-1],y);
}
}
return 0;
}