字符串KMP || POJ 2185 Milking Grid

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define SZ 10005
int nxt[SZ];
char s[SZ][80], tmp[80];
int r, c, k, cnt[80];//cnt相当于桶，记录每个长度出现多少次，如果一个长度出现r次即表示这个长度在每行都出现过，即为最终矩阵的宽
void get_next(char p[], int plen)
{
    int i = 0, j = -1;
    nxt[0] = -1;
    while(i < plen)
    {
        if(j == -1 || p[i] == p[j])
        {
            i++;
            j++;
            nxt[i] = j;
        }
        else j = nxt[j];
    }
    k = c - nxt[plen];//后缀
    cnt[k]++;
    if(k == c) return;
    for(int i = 0; i < nxt[plen]; i++)
        tmp[i] = p[i + plen - nxt[plen]];//把后缀拿出来，再kmp
    get_next(tmp, nxt[plen]);
    return;
}
void get_next_str(char p[][80], int plen)
{
    int i = 0, j = -1;
    nxt[0] = -1;
    while(i < plen)
    {
        if(j == -1 || strcmp(p[i], p[j]) == 0)//把每行看作一个整体
        {
            i++;
            j++;
            nxt[i] = j;
        }
        else j = nxt[j];
    }
    return;
}
int main()
{
    scanf("%d %d", &r, &c);
    for(int i = 0; i < r; i++)
        scanf(" %s", s[i]);
    for(int i = 0; i < r; i++)
        get_next(s[i], c);
    for(int i = 0; i < c; i++)
        if(cnt[i] == r)
        {
            c = i;//i长度出现了r次，即为最终的宽
            break;
        }
    for(int i = 0; i < r; i++)
        s[i][c] = ‘\0‘;//把剩下的列砍掉
    get_next_str(s, r);
    printf("%d\n", c * (r - nxt[r]));
    return 0;
}