HDU—4046 Panda （线段树）

题目：http://acm.hdu.edu.cn/showproblem.php?pid=4046

 

题意：给出一个字符串，统计这个字符串任意区间中“wbw”出现的次数。

规定两种操作，一是查询任意区间“wbw”出现次数；二是修改某一位置的字符。

 

分析：比较明显的线段树，单点更新，区间查询。

线段树记录的信息是区间中出现“wbw”字符的个数，线段树的叶子节点[i,i]记录字符串str中

str[i-2][i-1][i]是否是“wbw“。

对于字符的修改，这里就有要注意了。修改的位置pos, 可能对线段树中sum[pos-1]   sum[pos] 或sum[pos+1]

造成影响。这里要分情况。

 

代码

#include<cstdio>
#include<cstring>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int maxn=50010;
int n,m,sum[maxn<<2];
char str[maxn];
int num[maxn];
void pushup(int rt)
{
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
	if(l==r)
	{
		sum[rt]=num[l];
		return;
	}
	int m=(l+r)>>1;
	build(lson);
	build(rson);
	pushup(rt);
}
int query(int L,int R,int l,int r,int rt)
{
	if(L<=l&&r<=R)
	{
		return sum[rt];
	}
	int m=(l+r)>>1;
	int ans=0;
	if(L<=m) ans+=query(L,R,lson);
	if(R>m)  ans+=query(L,R,rson);
	return ans;
}
void update(int p,int val,int l,int r,int rt)
{
	if(l==r)
	{
		sum[rt]=val;
		return;
	}
	int m=(l+r)>>1;
	if(p<=m) update(p,val,lson);
	else 	 update(p,val,rson);
	pushup(rt);
}
int main()
{
	int t,T;
	scanf("%d",&T);
	for(int t=1;t<=T;t++)
	{
		printf("Case %d:\n",t);
		scanf("%d%d%s",&n,&m,str+1);
		memset(num,0,sizeof(num));
		for(int i=3;i<=n;i++)
			if(str[i-2]=='w'&&str[i-1]=='b'&&str[i]=='w')  num[i]=1;
		build(1,n,1);
		int k,a,b;
		char ch[5];
		while (m--)
		{
			scanf("%d",&k);
			if(k==0)
			{
				scanf("%d%d",&a,&b);
				a++;b++;   //字符串从1开始，所以下标都加1
				if(b-a<2)  printf("0\n");
				else       printf("%d\n",query(a+2,b,1,n,1));
			}
			else
			{
				scanf("%d%s",&a,ch);
				a++;
				if(ch[0]==str[a]) continue;//修改的和以前一样，这不用任何操作
				if(a>=3)
				{
					if(str[a-2]=='w'&&str[a-1]=='b'&&str[a]=='w')
						update(a,0,1,n,1);
					if(str[a-2]=='w'&&str[a-1]=='b'&&str[a]=='b')
						update(a,1,1,n,1);
				}
				if(a>=2&&a+1<=n)
				{
					if(str[a-1]=='w'&&str[a]=='b'&&str[a+1]=='w')
						update(a+1,0,1,n,1);
					if(str[a-1]=='w'&&str[a]=='w'&&str[a+1]=='w')
						update(a+1,1,1,n,1);
				}
				if(a+2<=n)
				{
					if(str[a]=='w'&&str[a+1]=='b'&&str[a+2]=='w')
						update(a+2,0,1,n,1);
					if(str[a]=='b'&&str[a+1]=='b'&&str[a+2]=='w')
						update(a+2,1,1,n,1);
				}
				str[a]=ch[0];
			}
		}
	}
	return 0;
}