11019 - Matrix Matcher--AC自动机

在一个字符矩阵里面找出模块矩阵出现了多少次？

处理模块行的时候注意，模块行会出现重复的情况，这样后插入的行就会覆盖先前的行，可以为每次插入的行弄一个链表。

参考：点击打开链接

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
#include <map>
using namespace std;
const int SIGMA_LETTER = 63;
const int MAX_NODE = 10000;
const int MAXN = 1010;
int dic[MAXN][MAXN];

struct AhoCorasick
{
    int ch[MAX_NODE][SIGMA_LETTER];
    int val[MAX_NODE];
    int f[MAX_NODE];
    int next[MAX_NODE];
    //int last[MAX_NODE];//suffix link
    int sz;
    void init()
    {
        sz=0;
        memset(next,0,sizeof(next));
        memset(val,0,sizeof(val));
        memset(ch[0],0,sizeof(ch[0]));
        memset(dic,0,sizeof(dic));
    }
    //返回单个字符的数字索引
    int idx(char c)
    {
        if(c>='a'&&c<='z')return c-'a';
        else if(c>='A' && c<='Z')return c-'A'+26;
        else return c-'0'+52;
    }
    void insert(char *w,int v)
    {
        int len =strlen(w);
        int now=0,c;
        for(int i=0;i<len;i++)
        {
            c= idx(w[i]);
            if(ch[now][c]==0)
            {
                ch[now][c] = ++sz;
                memset(ch[sz],0,sizeof(ch[sz]));
            }
            now = ch[now][c];
        }
        if(val[now])
        {
            next[v]=val[now];
        }
        val[now]=v;
    }
    void build_()
    {
        int u,c;
        queue<int> Q;
        for(int i =0;i<SIGMA_LETTER;i++)
        {
            u=ch[0][i];
            if(!u)continue;
            Q.push(u);f[u]=0;//last[u]=0;
        }
        while(!Q.empty())
        {
            u=Q.front();Q.pop();
            //依次为u节点的孩子匹配前缀指针
            for(int i=0;i<SIGMA_LETTER;i++)
            {
               c = ch[u][i];
               if(!c)continue;
               Q.push(c);
               int  v = f[u];
               while(v && !ch[v][i])  v=f[v];
               f[c] = ch[v][i];
              // last[c] = val[f[c]] ? f[c]:last[f[c]];
            }
        }
    }
    void find(char *T,int line)
    {
        int len =strlen(T);
        int now =0,c;
        for(int i=0;i<len;i++)
        {
            c = idx(T[i]);
            while(now && !ch[now][c]) now =f[now];
            now  = ch[now][c];
            if(val[now])mark(i,val[now],line);
        }
     }
     void mark(int col,int row,int line)
     {
        if(!row)return;
        if(line-row+1>0)
        dic[line-row+1][col]++;
        mark(col,next[row],line);
     }
};
int T,n,m,x,y;
char p[MAXN][MAXN];
AhoCorasick ac;
void solve1()
{
    int ans=0;
    if(x>n||y>m)
    {
        printf("0\n");
        return;
    }
    for(int i=1;i<=n;i++)
    {
        ac.find(p[i]+1,i);
    }
    for(int i =1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            //printf("%d ",dic[i][j]);
            if(dic[i][j]==x)
            ans++;
        }
       // printf("\n");
    }
    printf("%d\n",ans);
}

int main()
{
    freopen("data.in","r",stdin);
    char temp_str[110];
    scanf("%d",&T);
    while(T--)
    {
        ac.init();
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%s",p[i]+1);
        }
        scanf("%d%d",&x,&y);
        for(int i=1;i<=x;i++)
        {
            scanf("%s",temp_str);
            ac.insert(temp_str,i);
        }
        ac.build_();
        solve1();
    }
    return 0;
}

其实命运是自己选择的，此刻选择是懒惰还是踏实学习，注定未来当码农还是当厨师。