hdu 4749 Parade Show(KMP)

题目链接：hdu 4749 Parade Show

题目大意：学校要举办一场典礼，要从n个学生中选若干支队伍来，每支队伍有m个人。然后现在将学生的身高划分成k个等级，接着按照学生的顺序给出学生的身高等级，再然后，给出m个人的队伍要求的相对高度。然后要求从n个中间挑选连续的m个人，满足相对高度即可组成一支队伍。问说最多可以组成多少支队伍。

解题思路：KMP算法的变形，只不过在判断相等的时候有点难办，因为它算的是相对高度，就是同一支队伍中，对应比自己高的个数，相等的个数，比自己矮的个数都要和要求的相等，所以要开一个cx，cy数组，将KMP中判断相等写成一个函数。

#include <stdio.h>
#include <string.h>

const int N = 1e5+5;
int n, m, k, next
, x
, y
;
int cx
[30], cy
[30];

void init () {
int a;

memset(cx[0], 0, sizeof(cx[0]));
memset(cy[0], 0, sizeof(cy[0]));

for (int i = 1; i <= n; i++) {
memcpy(cx[i], cx[i-1], sizeof(cx[i-1]));
scanf("%d", &x[i]);
cx[i][x[i]]++;
}

for (int i = 1; i <= m; i++) {
memcpy(cy[i], cy[i-1], sizeof(cy[i-1]));
scanf("%d", &y[i]);
cy[i][y[i]]++;
}
}

bool judge (int p, int q, int (*a)[30], int (*b)[30], int tp, int tq) {
int s = 0;
for (int i = 1; i < tp; i++)
s += a[p][i];
for (int i = 1; i < tq; i++)
s -= (b[q][i] - b[q-p][i]);
return s == 0 && a[p][tp] == (b[q][tq] - b[q-p][tq]);
}

void getNext () {
int j = 0;
for (int i = 2; i <= m; i++) {
while (j > 0 && !judge (j+1, i, cy, cy, y[j+1], y[i]))
j = next[j];

if (judge (j+1, i, cy, cy, y[j+1], y[i]))
j++;
next[i] = j;
}
}

int KMP () {
int ans = 0;
getNext();

int j = 0;
for (int i = 1; i <= n; i++) {
while (j > 0 && !judge (j+1, i, cy, cx, y[j+1], x[i]))
j = next[j];

if (judge (j+1, i, cy, cx, y[j+1], x[i]))
j++;

if (j == m) {
ans++;
j = 0;
}
}
return ans;
}

int main () {
while (scanf("%d%d%d", &n, &m, &k) == 3) {
init ();
printf("%d\n", KMP());
}
return 0;
}