POJ 2185 KMP中next数组的应用

这道题一开始没明白意思，后来一查，是说给定一个字符矩阵，然后把能够构成这个字符矩阵的最小字符矩阵的字符个数给求出来。

解法是KMP中next数组的应用，先对每一行进行get_next，求出他们的最小公倍数；再对每一列进行get_next，求出最小公倍数。这两步都很好想，然后r * c即可。

坑爹的是，我wa了5、6次都是wa在get_next(char s*)这个函数上。教训就是，以后涉及字符串处理的函数，必须避开使用指针或字符串做函数参数，否则会wa出翔的……

Code:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

const int maxr = 11111;
const int maxc = 100;
int r, c;
char a[maxr][maxc], buf[maxr];
int next[maxr];

int get_next(int len)
{
     next[0] = -1;
     int i = 0, j = -1;
     while(i < len)
     {
             if(j == -1 || buf[i] == buf[j])
             {
                  i ++, j ++;
                  next[i] = j;
             }
             else
                  j = next[j];
     }
     return len - next[len];
}

int gcd(int a, int b)
{
    return b ? gcd(b, a % b) : a;
}

int lcm(int a, int b)
{
    return a / gcd(a, b) * b;
}

int main()
{
    while(~scanf("%d%d", &r, &c))
    {
             int i, j;
             for(i = 0; i < r; i ++)
                   scanf("%s", a[i]);
             int rr = 1;
             for(i = 0; i < r; i ++)
             {
                   for(j = 0; j < c; j ++)
                         buf[j] = a[i][j];
                   rr = lcm(rr, get_next(c));
                   if(rr > c)
                   {
                         rr = c;
                         break;
                   }
             }
             int cc = 1;
             for(i = 0; i < c; i ++)
             {
                   for(j = 0; j < r; j ++)
                         buf[j] = a[j][i];
                   cc = lcm(cc, get_next(r));
                   if(cc > r)
                   {
                         cc = r;
                         break;
                   }
             }
             printf("%d\n", cc * rr);
    }
}