XDU 1154 大黄的选票 KMP

题意：中文题...

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

求大神纠错。

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

代码：

/*
*  Author:      illuz <iilluzen[at]gmail.com>
*  Blog:        http://blog.csdn.net/hcbbt
*  File:        xdu1154.cpp
*  Create Date: 2013-11-27 23:18:07
*  Descripton:  kmp 
*/

#include <cstdio>
#include <cstdlib>
#include <cstring>

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;
}