UVa 11019 Matrix Matcher

题目链接

UVa 11019 Matrix Matcher

题解

这种二维化一维的方法并不少见但是我觉得我还是想不出来.

字符矩阵P可以先拆成很多行然后合成一个AC自动机,然后去匹配T的每一行,然后用cnt[i][j]表示以(i,j)为左上角这个大小为P的矩阵里有多少行与P匹配,那如果达到x了的话就是一个匹配点,当在T的第r行匹配到P的第i行(也就是AC自动机里面编为i的模板),最左边一列为c时那么cnt[r-i][c]应该++,(这里要求从0开始编号)注意判断r-i>=0;还有一个很重要的地方是模板可能会相同,那么在build trie的时候就需要搞一点事情,把每个节点串成一条链,因为模板编号是惟一的,所以可以用next[i]表示与i相同的下一个模板的编号,刚开始将所有trie节点的val置为-1.

代码

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;
const int maxx=100+10;
const int maxn=1000+10;
const int sigma_size=26;
const int maxnode=100*100+10;
int idx(char ch){
return ch-'a';
}
int read(){
int ret=0;char ch=getchar();
while(ch<'0'||ch>'9') ch=getchar();
for(;ch>='0'&&ch<='9';ch=getchar()) ret=ret*10+ch-'0';
return ret;
}

int n,m,x,y,next[maxn];
struct AC_auto{
int sz,val[maxnode];
int ch[maxnode][sigma_size];
inline void clear(){
sz=1;val[0]=-1;
memset(ch[0],0,sizeof(ch[0]));
memset(count,0,sizeof(count));
}
inline void insert(char *T,int v){
int n=strlen(T),u=0;
for(int i=0;i<n;i++)
{
int c=idx(T[i]);
if(!ch[u][c]){
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=-1;ch[u][c]=sz++;
}
u=ch[u][c];
}
next[v]=val[u];val[u]=v;
}
int f[maxnode],count[maxn][maxn];
inline void getfail(){
queue<int>q;
f[0]=0;
for(int i=0;i<sigma_size;i++)
{
int x=ch[0][i];
if(x) f[x]=0,q.push(x);
}
while(!q.empty())
{
int r=q.front();q.pop();
for(int c=0;c<sigma_size;c++)
{
int u=ch[r][c];
if(!u) {
ch[r][c]=ch[f[r]][c];continue;
}
q.push(u);
int v=f[r];
while(v&&!ch[v][c]) v=f[v];
f[u]=ch[v][c];
}
}
return ;
}
inline void deal(int cur,int j,int i){
for(;j!=-1;j=next[j])
if(cur>=j) count[cur-j][i]++;
}
inline void find(int cur,char *T){
int n=strlen(T),j=0;
for(int i=0;i<n;i++)
{
int c=idx(T[i]);
j=ch[j][c];
if(val[j]!=-1) deal(cur,val[j],i-y+1);
}
}
}ac;

char T[maxn][maxn],S[maxx][maxx];
inline void solve()
{
int ans=0;ac.clear();
n=read(),m=read();
rep(i,n) scanf("%s",T[i]);
x=read(),y=read();
rep(i,x) {
scanf("%s",S[i]);
ac.insert(S[i],i);
}
ac.getfail();
rep(i,n) ac.find(i,T[i]);
rep(i,n) rep(j,n) if(ac.count[i][j]==x) ans++;
printf("%d\n",ans);
return ;
}

int main()
{
int T;cin>>T;
while(T--) solve();
return 0;
}