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

时间:2023-12-09 18:54:01

【题目链接】 http://acm.hdu.edu.cn/showproblem.php?pid=5716

【题目大意】

  给出一个字符串,找出其中所有的符合特定模式的子串位置,符合特定模式是指,该子串的长度为n,并且第i个字符需要在给定的字符集合Si中

【题解】

  这种串与字符集的匹配称为柔性字符串匹配,采用ShiftAnd的匹配方法。

  bt[i]表示字符i允许在哪些位置上出现,我们将匹配成功的位置保存在dp中,那么就可以用dp[i]=dp[i-1]<<1&bt[s[i]]来更新答案了

  利用滚动数组和bitset可以来优化这样的运算,当一个位置的匹配在更新的过程中没有丢失,即始终在特定模式中直到定长,那么这个位置就是成功匹配位

  复杂度为O(nm/64)

【代码】

#include <cstdio>
#include <bitset>
#include <cstring>
using namespace std;
const int M=510,N=2000010,L=65,U=256;
bitset<M> dp[2],bt[U];
int n,m,id[U],cnt,l;
char s[N],t[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 n,int m){
int cur=1,f=0;
dp[0].reset(); dp[0].set(0);
for(int i=1;i<=n;i++,cur^=1){
dp[cur]=dp[cur^1]<<1&bt[id[s[i]]];
dp[cur].set(0);
if(dp[cur][m])printf("%d\n",i-m+(f=1));
}if(!f)puts("NULL");
}
int main(){
init();
while(gets(s+1)){
n=strlen(s+1);
scanf("%d",&m);
for(int i=1;i<=cnt;i++)bt[i].reset();
for(int i=1;i<=m;i++){
scanf("%d %s",&l,t);
for(int j=0;j<l;j++)bt[id[t[j]]].set(i);
}ShiftAnd(n,m);gets(s);
}return 0;
}