首页 > 代码库 > BZOJ 2002 HNOI2010 弹飞绵羊 分块

BZOJ 2002 HNOI2010 弹飞绵羊 分块

题目大意及LCT版本题解:见 http://blog.csdn.net/popoqqq/article/details/38849471

今天手滑用分块又重写了一遍这道题0.0 分块就是短啊

将弹簧分为√n块

对于每个弹簧 我们记录一下从这个弹簧出发直到弹到块外为止的弹跳次数及落点

查询沿着落点弹到出去为止 修改从块开始到这个点为止修改一遍

这样修改和查询都是O(√n)的

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 200200
using namespace std;
int n,m,ans,block,a[M];
int pos[M],times[M];
int main()
{
	int i,j,p,x,y;
	cin>>n;
	for(i=0;i<n;i++)
		scanf("%d",&a[i]);
	block=static_cast<int>(sqrt(n)+1e-7);
	for(i=n-1;~i;i--)
	{
		int temp=i+a[i];
		if(temp>=n)
			pos[i]=-1,times[i]=1;
		else if(temp>=(i/block+1)*block)
			pos[i]=temp,times[i]=1;
		else
			pos[i]=pos[temp],times[i]=times[temp]+1;
	}
	cin>>m;
	for(i=1;i<=m;i++)
	{
		scanf("%d",&p);
		if(p==1)
		{
			scanf("%d",&x);
			ans=0;
			for(j=x;~j;j=pos[j])
				ans+=times[j];
			printf("%d\n",ans);
		}
		else
		{
			scanf("%d%d",&x,&y);
			a[x]=y;
			for(j=x;j>=x/block*block;j--)
			{
				int temp=j+a[j];
				if(temp>=n)
					pos[j]=-1,times[j]=1;
				else if(temp>=(j/block+1)*block)
					pos[j]=temp,times[j]=1;
				else
					pos[j]=pos[temp],times[j]=times[temp]+1;
			}
		}
	}
}


BZOJ 2002 HNOI2010 弹飞绵羊 分块