poj2185(kmp)

题目链接: http://poj.org/problem?id=2185

 

题意: 给出一个矩形字符块, 求能组成该矩形块的最小子矩形块.

 

思路: 将每行字符串当作一个字符,那么原矩形字符块就成了一列字符,其最小循环节即为所求矩形块的宽；

将每列字符串当作一个字符,那么原矩形字符块就成了一行字符,其最小循环节即为所求矩形的长.

 

代码:

#include <iostream>
#include <stdio.h>
using namespace std;

const int MAXN = 1e5 + 10;
int nxt1[MAXN], nxt2[80];
char str[MAXN][80];
int n, m;

bool equal1(int x, int y){//判断两行字符串是否相等
    for(int i = 0; i < m; i++){
        if(str[x][i] != str[y][i]) return false;
    }
    return true;
}

bool equal2(int x, int y){//判断两列字符串是否相等
    for(int i = 0; i < n; i++){
        if(str[i][x] != str[i][y]) return false;
    }
    return true;
}

void get_nxt1(void){ //将每行字符串当作一个字符,求矩形字符块变成字符串后的nxt数组
    int i = 0, j = -1;
    nxt1[0] = -1;
    while(i < n){
        if(j == -1 || equal1(i, j)) nxt1[++i] = ++j;
        else j = nxt1[j];
    }
}

void get_nxt2(void){ //将每列字符串当作一个字符,求矩形字符块变成字符串后的nxt数组
    int i = 0, j = -1;
    nxt2[0] = -1;
    while(i < m){
        if(j == -1 || equal2(i, j)) nxt2[++i] = ++j;
        else j = nxt2[j];
    }
}

int main(void){
    scanf("%d%d", &n, &m);
    for(int i = 0; i < n; i++){
        scanf("%s", str[i]);
    }
    get_nxt1();
    get_nxt2();
    int cnt1 = n - nxt1[n];
    int cnt2 = m - nxt2[m];
    printf("%d\n", cnt1 * cnt2);
    return 0;
}