XDU 1154 大黄的选票 KMP

来源：互联网发布：mmd 绅士动作数据编辑：程序博客网时间：2024/04/28 08:28

题意：中文题...

别人都是用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;} elsej = f[j];}}int max(int a, int b) {return a > b ? a : b;}void solve() {// s1 kmpmemset(rec, 0, sizeof(rec));int lp = strlen(s1), lt = strlen(s), lm;// lm 为s1 s2中较长的长度，因为最短的符合的字串可能是lmlm = 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++;} elsej = f[j];if (j == lp) {// 记录s1出现的位置rec[i - lp + 1] = 1;j = f[j];}}// 计数求和，rec为前i个出现了几个s1for (int i = 1; i <= lt; i++) {rec[i] += rec[i - 1];}// s2 kmpint lp2 = strlen(s2), ans = 0;getVal(s2, lp2);i = 0;   j = 0;while (i < lt) {if (j == -1 || s[i] == s2[j]) {i++;j++;} elsej = 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;}