codeforces 313B Ilya and Queries

线段树简单查询即可,注意所开数组大小

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#define ll long long
#define ls rt<<1
#define rs ls1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define eps (1e-9)
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define MOD 1000000007
#define inf 100000007
#define pi acos(-1.0)
#define for(i,a,b) for(int i=(a);i<(b);i++)
#define M 100000+5
struct node{
int x,y,n;
}r[3*M];
char s[M]; 
int cnt,len;
void Init(int ld,int rd,int p){
r[p].x=ld,r[p].y=rd;
if(ld==rd){
if(ld<len)r[p].n=(s[ld]==s[ld+1]?1:0);
return;
}
int mid=(ld+rd)/2;
Init(ld,mid,p*2);Init(mid+1,rd,p*2+1);
r[p].n=r[p*2].n+r[p*2+1].n;
// printf("%d %d %d %d\n",p,r[p].x,r[p].y,r[p].n);
}
void query(int ld,int rd,int p){
if(ld<=r[p].x&&rd>=r[p].y){
cnt+=r[p].n;
return;
}
int mid=(r[p].x+r[p].y)/2;
if(rd<=mid)query(ld,rd,p*2);
else if(ld>=mid+1)query(ld,rd,p*2+1);
else{
query(ld,mid,p*2);
query(mid+1,rd,p*2+1);
}
}
int main(){
    int m,i,j;
    while(~scanf("%s",s+1)){
memset(r,0,sizeof(r));
len=strlen(s+1);
Init(1,len,1);
    scanf("%d",&m);
    while(m--){
scanf("%d %d",&i,&j);
cnt=0;
query(i,j-1,1);
printf("%d\n",cnt);
}
}
    return 0;
}

你可能感兴趣的:(线段树,codeforces)