题目大意:给一个只有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]);
}
}