[POI2011]LIZ-Lollipop

[POI2011]LIZ-Lollipop

给一个只有1和2的序列，每次询问有没有一个子串的和为x。

解：

这道题需要观察性质。
本来想的前缀和做法，详情可以参见洛谷的题解。
然而大佬表示有一种更简单的做法。



由于只有1和2，我们可以发现只要$x$可以被表示出来那么比$x$小且与$x$奇偶性相同的数都可以被表示出来。为什么？
假设当前数列的两端：
1,1：只需要左右都去掉一个1
1,2：去掉2
2,1：同上
2,2,：任意去掉一个2
所以我们只需要找到最大能表示的奇偶数就行。
把所有数加起来可以得到一个和。然后找到左右两边第一个1，看哪个离端点最近，就去掉哪边。
然后开一个大数组，记录一下所有数的表示结果。然后就可以$O(1)$查询了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

int a[1000005];
int ans[2000005];
int li[2000005];
int ri[2000005];
char s[2000005];
int n,m,t;
int num;

int main()
{
    scanf("%d%d",&n,&m);
    scanf("%s",s+1);
    for(int i=1;i<=n;i++){
      if(s[i]=='T') a[i]=2;
      else a[i]=1;
    }
    for(int i=1;i<=n;i++)
      num+=a[i];
    int t1=1,t2=n;
    while(a[t1]==2&&t1<=n) t1++;
    while(a[t2]==2&&t2>=1) t2--;
    int l=1,r=n,t=num;
    while(l<=r){
        li[t]=l;ri[t]=r;
        if(a[l]+a[r]==2) l++,r--;
        else if(a[l]+a[r]==3){
            if(a[l]==2) l++;
            else r--;
        }
        else if(a[l]+a[r]==4) l++;
        t-=2;
    }
    if(t1-1>n-t2) t=num-(n-t2)*2-1,l=1,r=t2-1;
    else t=num-(t1-1)*2-1,l=t1+1,r=n;
    while(l<=r){
        li[t]=l;ri[t]=r;
        if(a[l]+a[r]==2) l++,r--;
        else if(a[l]+a[r]==3){
            if(a[l]==2) l++;
            else r--;
        }
        else if(a[l]+a[r]==4) l++;
        t-=2;
    }
    for(int i=1;i<=m;i++)
    {
        int x;
        scanf("%d",&x);
        if(li[x]==0&&ri[x]==0) printf("NIE\n");
        else printf("%d %d\n",li[x],ri[x]);
    }
}