HDU 4749 Parade Show Kmp

  這題用的是kmp算法，題意是給出兩個隊列，每個隊列的人的高度，讓你求隊列二相對於隊列一高低趨勢匹配的數目總共有多少。高低趨勢匹配就是2 4 2 4 2；1 2 1 2 1；

2 比 4 小；1 比 2 小，這就匹配了，一整個隊列都符合，算一個。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#define maxn 100000
#define mem(a) memset(a, 0, sizeof(a))
using namespace std;

char ch1[maxn + 5], ch2[maxn + 5];
int next1[maxn + 5], n, m, ans;

void get_next(int len)
{
	int i, j = 0;
	next1[1] = 0;
	for(i = 2;i <= len;i++)
	{
		if(ch2[j + 1] != ch2[i]&&j > 0)
		j = next1[j];
		if(ch2[j + 1] == ch2[i])
		j++;
		next1[i] = j;
	}
	return;
}

bool juge(int a, int b)
{
    if(ch1[a] > ch1[a - 1]&&ch2[b] > ch2[b - 1])
	return true;	
	if(ch1[a] < ch1[a - 1]&&ch2[b] < ch2[b - 1])
	return true;
	if(ch1[a] == ch1[a - 1]&&ch2[b] == ch2[b - 1])
	return true;
	return false;
}

void get_ans(int len)
{
	int i, j = 1;
	for(i = 1;i < len;i++)
	{
		if(juge(i + 1, j + 1))
		j++;
		else if(!juge(i + 1, j + 1))
		{
			j = next1[j] + 1;
		}
		if(j == m)
		{
			ans++;
			j = 1;
			i++;
		}
	}
}

int main(int argc, char *argv[])
{
	int k, i, a;
	while(scanf("%d%d%d", &n, &m, &k) != EOF)
	{
		mem(ch1);
		mem(ch2);
		mem(next1);
		ans = 0;
		for(i = 1; i <= n;i++)
		{
			scanf("%d", &a);
			ch1[i] = a + 48;
		}
		for(i = 1;i <= m;i++)
		{
			scanf("%d", &a);
			ch2[i] = a + 48;
		}
		if(m == 1)
		{
		    printf("%d\n", n);
		    continue;
		}
		get_next(m);
		get_ans(n);
		printf("%d\n", ans);
	}
	return 0;
}