题目:https://www.lydsy.com/JudgeOnline/problem.php?id=2002
分块或者LCT。目前只会分块。
暴力的话,要么修改O(n)(倒序修改即可!),查询O(1);要么修改O(1),查询O(n)。
分块就是折中一下,仅修改本块中的点,查询时在一块上仅停留一下(每个点记录它到下一块的步数和位置);
这样修改和查询都是O(根号n),就能通过了。
可以倒序节省时间。最好记录一下块的标号,比较方便。
#include<iostream> #include<cstdio> #include<cstring> #include<cmath> using namespace std; const int N=2e5+5; int n,a[N],m,c[N],ps[N],bh[N],base; void solve(int k) { int x=k+a[k]; if(bh[x]==bh[k])ps[k]=ps[x],c[k]=c[x]+1; else ps[k]=x,c[k]=1; } int main() { scanf("%d",&n);base=sqrt(n); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); bh[i]=(i-1)/base+1; } for(int i=n;i;i--)solve(i); scanf("%d",&m);int x,y,z; while(m--) { scanf("%d%d",&x,&y);y++; if(x==1) { int ans=0; while(y<=n)ans+=c[y],y=ps[y]; printf("%d ",ans); } else{ scanf("%d",&z);a[y]=z;int l=(bh[y]-1)*base+1; for(int i=y;i>=l;i--)solve(i); } } return 0; }