http://cnc.boj.me/onlinejudge/newoj/showProblem/show_problem.php?problem_id=217
其他人都说用线段树,可是自己敲着敲着就没想法,只好转向树状数组,发现可以实现,貌似比别人的线段树快。
方法:一直维护wbw中第一个w,把它记为1,然后整个数组更新。求和比较头疼,第一是下标要加1,第二是很容易求错。
后来想了想,发现只要sum(r-2)-sum(l-1),其中l和r都转化为以1为开始下标。
#include <map>
#include <set>
#include <list>
#include <queue>
#include <deque>
#include <stack>
#include <string>
#include <time.h>
#include <cstdio>
#include <math.h>
#include <iomanip>
#include <cstdlib>
#include <limits.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
#define LL long long
#define PI acos(-1.0)
#define MAX INT_MAX
#define MIN INT_MIN
#define FRE freopen("a.txt","r",stdin)
#define N 50005
#define MOD 10007
int c[N];
char str[N];
int lowbit(int t){
return t&(-t);
}
void update(int pos,int v){
while(pos<N){
c[pos]+=v;
pos+=lowbit(pos);
}
}
int sum(int pos){
int tot=0;
while(pos){
tot+=c[pos];
pos-=lowbit(pos);
}
return tot;
}
int main(){
int tt;
scanf("%d",&tt);
int ca=1;
while(tt--){
int n,m;
scanf("%d%d",&n,&m);
int i,j;
scanf("%s",str);
memset(c,0,sizeof(c));
for(i=0;i+2<n;i++){
if(str[i]=='w' && str[i+1]=='b' && str[i+2]=='w'){
update(i+1,1);
}
}
printf("Case %d:\n",ca++);
while(m--){
int a;
scanf("%d",&a);
if(!a){
int s,t;
scanf("%d%d",&s,&t);
if(t-s+1<=2)printf("%d\n",0);
else{
int l=s+1;
int r=t+1;
printf("%d\n",sum(r-2)-sum(l-1));
}
}
else{
int pos;
char cc;
scanf("%d %c",&pos,&cc);
int s;
int t;
s=(pos-2>=0) ? pos-2 : 0 ;
t=(pos+2<n) ? pos+2 : n-1 ;
for(i=s;i+2<=t;i++)
if(str[i]=='w' && str[i+1]=='b' && str[i+2]=='w'){
update(i+1,-1);
}
str[pos]=cc;
for(i=s;i+2<=t;i++)
if(str[i]=='w' && str[i+1]=='b' && str[i+2]=='w'){
update(i+1,1);
}
}
}
}
return 0;
}