HDU 5716 带可选字符的多字符串匹配 ShiftAnd

不能用cin或者scanf读入，因为题目给的字符串中包含空格等乱七八糟的东西，只能用gets读入，最后用getchar吃一个换行符。



#include<iostream>
#include<queue>
#include<algorithm>
#include<stdlib.h>
#include<stdio.h>
#include<iomanip>
#include<bitset>
#include<string.h>

using namespace std;

const int MAXM=2000005;
const int MAXN=505;
const int L=65;
const int U=256;

bitset<MAXN> F[2],B[U];
int N,M,Id[U],Cnt,K;
char Str[MAXM],Temp[L];

void init()
{
    Cnt=0;
    for (int i='0';i<='9';i++) Id[i]=++Cnt;
    for (int i='A';i<='Z';i++) Id[i]=++Cnt;
    for (int i='a';i<='z';i++) Id[i]=++Cnt;
}

void ShiftAnd()
{
    int cur=1,flag=0;
    F[0].reset();
    F[0].set(0);
    for (int i=1;i<=M;i++,cur^=1)
    {
        F[cur]=F[cur^1]<<1&B[Id[Str[i]]];
        F[cur].set(0);
        if (F[cur][N])
            printf("%d\n",i-N+(flag=1));
    }
    if (!flag)
        printf("NULL\n");
}

int main()
{
    init();
    while (gets(Str+1))
    {
        M=strlen(Str+1);
        scanf("%d",&N);
        for (int i=1;i<=Cnt;i++)
            B[i].reset();
        for (int i=1;i<=N;i++)
        {
            scanf("%d %s",&K,Temp);
            for (int j=0;j<K;j++)
                B[Id[Temp[j]]].set(i);
        }
        ShiftAnd();
        getchar();
    }
    return 0;
}