为什么这样就过了,比赛的时候怎么改都WA了,我只不过把条件改得更严格一点,把多余的判断去掉而已;
题意:给你一个字符串,由'w' 和 'b' 组成,对该字符串有俩个操作,
当输入为0时,询问该区间[a,b] 内有多少个串 为 "wbw";
当输入为0时,将下标为k的字符改为输入的字符;
注意:这里的a,b,k表示的都是字符串的下标,也就是取值范围为[0,n-1];
分析:
对这种区间修改或询问的题目,首先想到的就是线段树或者树状数组了,不过还是线段树熟悉一点,我用线段树做的 ,关键还是构造模型,区间内每一个点的值表示的是什么?
我是这样表示的,每一个点表示的是以该下标结束的长度为3的子串是否为"wbw",若是,则等于1,否则等于0,保存在数组a[]中。
所以,当进行区间询问时,对[a,b]的询问则应该改为对区间[a+2,b],当然这里对a+2,和b的大小还是要讨论的;
当进行点修改时,则可能要修改三个点,因为改动一个字符时,会影响到三个长度为3的子串,所以有可能进行三次修改(之前在这里对条件的判断似乎不太严格,一直WA);
这样问题就解决了;
#include<iostream> #include<string> #define maxn 50010 #define lson l , m , rt << 1 #define rson m + 1 , r , rt << 1 | 1 using namespace std; int sum[maxn*4]; char str[maxn]; int a[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]=a[l]; return ; } int m = (l + r) >> 1; build(lson); build(rson); PushUP(rt); } void update(int p,int add,int l,int r,int rt) { if (l == r) { sum[rt] = add; return ; } int m = (l + r) >> 1; if (p <= m) update(p , add , lson); else update(p , add , 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 ret = 0; if (L <= m) ret += query(L , R , lson); if (R > m) ret += query(L , R , rson); return ret; } int main() { int cas,T=0,n,m,r,l,c; char st; scanf("%d",&cas); while(cas--) { scanf("%d %d",&n,&m); scanf("%s",str); memset(a,0,sizeof(a)); printf("Case %d:\n",++T); for(int i=2;i<n;i++)//标记以该下标结尾的长度为3的子串是否符合要求 { if(str[i]!='w') a[i]=0; else if(str[i-1]=='b'&&str[i-2]=='w') a[i]=1; } build(0,n,1); while(m--) { scanf("%d",&c); if(c==0) { scanf("%d %d",&l,&r); l+=2; if(l>r) printf("0\n"); else printf("%d\n",query(l,r,0,n,1)); } else { scanf("%d %c",&l,&st); if(str[l]==st)//若要就该的字符跟原先相同,则不用修改了 continue; str[l]=st; if(l>=2&&str[l-2]=='w'&&str[l-1]=='b'&&str[l]=='w') { if(a[l]==0)//若改之前是不符合要求的串,则更新 update(l,1,0,n,1); a[l]=1; } else if(l>=2&&a[l]==1)//若改之前是符合要求的串,则更新 { update(l,0,0,n,1); a[l]=0; } if(l>=1&&str[l]=='b'&&str[l-1]=='w'&&str[l+1]=='w') { if(a[l+1]==0) update(l+1,1,0,n,1); a[l+1]=1; } else if(l<n-1&&l>=1&&a[l+1]==1) a[l+1]=0,update(l+1,0,0,n,1); if(str[l]=='w'&&str[l+1]=='b'&&str[l+2]=='w') { if(a[l+2]==0) update(l+2,1,0,n,1); a[l+2]=1; } else if(l<n-2&&a[l+2]==1) a[l+2]=0,update(l+2,0,0,n,1); } } } return 0; }