POJ 2185 Milking Grid（KMP）

最新推荐文章于 2022-02-04 15:17:36 发布

原创最新推荐文章于 2022-02-04 15:17:36 发布 · 579 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM #KMP

KMP 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种基于KMP算法的二维字符串匹配方法，通过将传统的一维字符比较扩展到二维的行或列比较，实现了对矩阵形式文本的有效搜索。文章提供了完整的C++实现代码，并详细解释了如何为行和列构建next数组来加速匹配过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

解题思路：

算是一个多维的KMP，其实是一样的，不过把1个字符的比较改成一行或一列字符的比较，对行和列使用两次KMP，最后乘起来就可以了。

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#define LL long long
#define FOR(i,x,y) for(int i=x;i<=y;i++)
using namespace std;
const int maxn = 10000 + 10;
char s[maxn][100];
int rnext[maxn];
int cnext[maxn];
int R, C;
bool rsame(int x, int y)
{
    for(int i=0;i<C;i++)
    {
        if(s[x][i] != s[y][i])
            return 0;
    }
    return 1;
}
bool csame(int x, int y)
{
    for(int i=0;i<R;i++)
    {
        if(s[i][x] != s[i][y])
            return 0;
    }
    return 1;
}
int main()
{
    while(scanf("%d%d", &R, &C)!=EOF)
    {
        for(int i=0;i<R;i++)
            scanf("%s", s[i]);
        rnext[0] = 0, rnext[1] = 0;
        for(int i=1;i<C;i++)
        {
            int j = rnext[i];
            while(j && !csame(i,j)) j = rnext[j];
            rnext[i+1] = (csame(i,j)) ? j + 1 : 0;
        }
        /*for(int i=0;i<=C;i++)
            printf("%d ", rnext[i]);
        printf("\n");*/
        cnext[0] = 0, cnext[1] = 0;
        for(int i=1;i<R;i++)
        {
            int j = cnext[i];
            while(j && !rsame(i, j)) j = cnext[j];
            cnext[i+1] = (rsame(i, j)) ? j + 1 : 0;
        }
        /*for(int i=0;i<=R;i++)
            printf("%d ", cnext[i]);
        printf("\n");*/
        printf("%d\n", (R - cnext[R]) * (C - rnext[C]));
    }
}