bzoj 2217 [Poi2011]Lollipop 乱搞 贪心

2217: [Poi2011]Lollipop

Time Limit: 15 Sec  Memory Limit: 64 MBSec  Special Judge
Submit: 383  Solved: 159
[Submit][Status][Discuss]

Description

有一个长度为n的序列a1,a2,...,an。其中ai要么是1("W")，要么是2("T")。
现在有m个询问，每个询问是询问有没有一个连续的子序列，满足其和为q。

Input

第一行n，m (1<=n,m<=1000000)
第二行这个序列，起始编号为1，终止编号为n
下面每行一个询问q，询问有没有一个连续的子序列，满足其和为q (1<=q<=2000000)

Output

对于每个询问，输出一行，如果有，输出这个序列的起点和终点(如果有多个输出任意一个)；如果没有，输出“NIE”。

Sample Input

5 3
TWTWT
5
1
7

Sample Output

1 3
2 2
NIE

HINT

尚无SPJ，请不要提交

 

这题有一个性质，如果存在一个连续的序列和为k，那么在前缀和中一定存在k或者k+1

为什么？

可以证明，如果l-r和为k，那么后面部分，可以不断减去，前面部分不断加入，如果相差为2，那么后面部分减去一个1，或者2，使其

差<2就可以了，这样是保证了这个性质。

所以对于这道题，对于k判断前缀和十分有k，有的话直接输出，否则判断是否有k+1，如果有的话，就记录前面二的个数，后面哪个位置是1，

然后搞一搞就好了。

#pragma GCC optimize(2)
#pragma G++ optimize(2)
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>

#define N 27
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

#define MAXN 1000010
#define MAXM 1010
#define INF 1000000000
#define MOD 1000000007
#define eps 1e-8
#define ll long long

int n,m;
char a[MAXN];
int s[MAXN],nxt[MAXN];
int v[MAXN<<1];
int main()
{
    int i,x;
    scanf("%d%d%s",&n,&m,a+1);
    for(i=1;i<=n;i++)
    {
        s[i]=s[i-1]+1+(a[i]=='T');
        v[s[i]]=i;
    }
    x=n+1;
    for(i=n;i;i--)
    {
        if(a[i]=='W')x=i;
        nxt[i]=x-i;
    }
    while(m--)
    {
        scanf("%d",&x);
        if(v[x])printf("%d %d
",1,v[x]);
        else if(v[x+1])
        {
            int l=1,r=v[x+1];
            if(nxt[l]<nxt[r])
            {
                r+=nxt[l];
                l+=nxt[l]+1;
                printf("%d %d
",l,r);
            }else if(r+nxt[r]<=n)
            {
                l+=nxt[r];
                r+=nxt[r];
                printf("%d %d
",l,r);
            }else printf("NIE
");
        }else printf("NIE
");
    }
}