XDU 1154 大黄的选票 KMP - c++编程基础

题意：中文题...

别人都是用KMP+DP的，我偷懒了，直接kmp一遍s1并记录它出现的位置，然后kmp后缀s2，看它前面有几个s1，这样就能构成几个合法字串了，但是一直wa，测试了好多组数据都没错的。

求大神纠错。

感谢shuangde大神提供数据，最后发现是lenp2传参错误，导致next()出错，我还以为我的getVal数组敲错了呢，前面几个都这样写的...

代码：

/* 
*  Author:      illuz  
*  Blog:        http://blog.csdn.net/hcbbt 
*  File:        xdu1154.cpp 
*  Create Date: 2013-11-27 23:18:07 
*  Descripton:  kmp  
*/  
  
#include   
#include   
#include   
  
const int MAXN = 1e4 + 10;  
  
char s[MAXN], s1[MAXN], s2[MAXN];  
int rec[MAXN], f[MAXN];  
  
void getVal(char* P, int l) {  
    int i = 0, j = -1;  
    f[0] = -1;  
    while (i < l) {  
        if (j == -1 || P[i] == P[j]) {  
            i++;  
            j++;  
            f[i] = j;  
        } else  
            j = f[j];  
    }  
}  
  
int max(int a, int b) {  
    return a > b   a : b;  
}  
  
void solve() {  
    // s1 kmp  
    memset(rec, 0, sizeof(rec));  
    int lp = strlen(s1), lt = strlen(s), lm;  
    // lm 为s1 s2中较长的长度，因为最短的符合的字串可能是lm  
    lm = max(lp, strlen(s2));  
    if (lm >




 lt) {  
        puts("0");  
        return;  
    }  
    getVal(s1, lp);  
    int i = 0, j = 0;  
    while (i < lt) {  
        if (j == -1 || s[i] == s1[j]) {  
            i++;  
            j++;  
        } else  
            j = f[j];  
        if (j == lp) {  
            // 记录s1出现的位置  
            rec[i - lp + 1] = 1;  
            j = f[j];  
        }  
    }  
    // 计数求和，rec为前i个出现了几个s1  
    for (int i = 1; i <= lt; i++) {  
        rec[i] += rec[i - 1];  
    }  
    // s2 kmp  
    int lp2 = strlen(s2), ans = 0;  
    getVal(s2, lp2);  
    i = 0;  
    j = 0;  
    while (i < lt) {  
        if (j == -1 || s[i] == s2[j]) {  
            i++;  
            j++;  
        } else  
            j = f[j];  
        if (j == lp2) {  
            // 找到可能构成合法字串的位置，计算这个后缀能构成几个合法串  
            ans += rec[i - lm + 1];  
            j = f[j];  
        }  
    }  
    printf("%d\n", ans);  
}  
  
int main() {  
    while (~scanf("%s%s%s", s, s1, s2)) {  
        solve();  
    }  
    return 0;  
}