BZOJ2217: [Poi2011]Lollipop

题目大意：给一个只有1和2的序列，每次询问有没有一个子串的和为x

先预处理出前缀和，然后对于每个询问，假如恰好有一个前缀和=x，那么就直接输出

否则一定有一个前缀和为x+1的地方，这时只要开头是1或者开头是2后面是1都可以直接出解..

也就是说不能直接出解的情况只是前后都是2，所以预处理出每个点向后最多有多少个连续的2，这道题就可以做了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000010
using namespace std;
char s[N];
int pre[N],L[N];
struct ppp{int w,num;}b[N<<1];
bool cmp(ppp x,ppp y){return x.w<y.w;}
int ansl[N<<1],ansr[N<<1];
int main()
{
	int n,m;
	scanf("%d%d%s",&n,&m,s+1);
	int i,j,x,tmp1,tmp2;
	for(i=1;i<=n;i++)
	pre[i]=pre[i-1]+(s[i]=='W'?1:2);
	for(i=1;i<=m;i++)
	{
		scanf("%d",&b[i].w);
		b[i].num=i;
	}
	L[n]=0;
	for(i=n-1;i>=0;i--)
	{
		if(s[i+1]=='T') L[i]=L[i+1]+1;
		else L[i]=0;
	}
	sort(b+1,b+m+1,cmp);
	j=0;
	for(i=1;i<=m;i++)
	{
		if(b[i].w>pre[n]) break;
		while(pre[j]<b[i].w) j++;
		if(pre[j]==b[i].w) ansl[b[i].num]=1,ansr[b[i].num]=j;
		else
		{
			x=min(L[j],L[0]);
			tmp1=x+1;tmp2=x+j;
			ansl[b[i].num]=tmp1+1;
			if(s[tmp1]=='W') ansr[b[i].num]=tmp2;
			else if(s[tmp2+1]=='W') ansr[b[i].num]=tmp2+1;
		}
	}
	for(i=1;i<=m;i++)
	{
		if(!ansr[i]) puts("NIE");
		else printf("%d %d\n",ansl[i],ansr[i]);
	}
}