题目大意:给定一个由1和2组成的序列,多次询问是否存在一个区间满足区间和=x
如果x>sum显然无解
如果存在一个前缀和为x则直接输出
否则一定存在一个前缀和[1,i]等于x+1
然后我们将左右端点同时右移 显然如果某一时刻a[l]=1或者a[r+1]=1那么我们就找到解了
记录exti表示从i开始有多少个连续的2
如果ext1<exti,那么解为[1+ext1+1,i+ext1]
如果ext1≥exti且i+exti≠n+1,那么解为[1+exti,i+exti]
否则无解
O(n)预处理所有答案即可
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 2002002
using namespace std;
int n,m,sum;
char s[M>>1];
int ext[M],l[M],r[M];
int main()
{
int i,x;
cin>>n>>m;
scanf("%s",s+1);
for(i=n;i;i--)
{
ext[i]=ext[i+1]+1;
if(s[i]==‘W‘)
ext[i]=0;
}
for(i=1;i<=n;i++)
{
sum+=s[i]==‘W‘?1:2;
l[sum]=1;r[sum]=i;
if(s[i]==‘T‘)
{
if(ext[1]<ext[i])
l[sum-1]=ext[1]+2,r[sum-1]=i+ext[1];
else if(i+ext[i]!=n+1)
l[sum-1]=1+ext[i],r[sum-1]=i+ext[i];
}
}
for(i=1;i<=m;i++)
{
scanf("%d",&x);
if( x>sum || !l[x] )
puts("NIE");
else
printf("%d %d\n",l[x],r[x]);
}
return 0;
}
时间: 2024-10-07 02:46:30