bzoj1729 [Usaco2005 dec]Cow Patterns 牛的模式匹配 kmp+乱搞

题意：给定一个顺序序列表示奶牛的排列顺序，定义串a=b当且仅当a的每位的排名=b的每位的排名。（注意排名仅相对于当前区间）。

一开始想的是一边kmp一边维护排名，但是有点复杂，然后学习了一下更为简单的一种做法。

设rank[i][j]表示前i位数字j出现的次数，那么我们在kmp的时候就可以一边跑一边看排名是否相同，具体的话，是这样：

假设kmp到当前第i位，已经匹配了j位，那么我们要求i,j排名相同，那么怎么求排名呢。

首先算出在当前范围内小于s2[i]的个数，然后算出i位置上的s2[i]已经出现了几次，这就是i的排名，j同理，然后这些都可暴力算出来，S<=25完全不会超。

细节不懂看代码。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N = 1e5 + 5;
char s1
, s2
;
int rank1
[30], rank2
[30];
int next
, ans
;
int tot, n, m, s;
inline void get()
{
int k = 0, i = 1;
while (i <= m)
{
int x1 = 0, y1 = 0, x2 = 0, y2 = 0;
fo(j, 1, s2[i] - 1)
x1 += rank2[i][j] - rank2[i - k][j];
y1 = rank2[i][s2[i]] - rank2[i - k][s2[i]];

fo(j, 1, s2[k] - 1)
x2 += rank2[k][j];
y2 = rank2[k][s2[k]];
if (k == 0 || (x1 == x2&&y1 == y2))i++, k++, next[i] = k;
else k = next[k];
}
}
inline void kmp()
{
int i = 1, j = 1;
while (i <= n)
{
int x1 = 0, x2 = 0, y1 = 0, y2 = 0;
fo(k, 1, s1[i] - 1)
x1 += rank1[i][k] - rank1[i - j][k];
y1 = rank1[i][s1[i]] - rank1[i - j][s1[i]];
fo(k, 1, s2[j] - 1)
x2 += rank2[j][k];
y2 = rank2[j][s2[j]];
if (!j || (x1 == x2&&y1 == y2))j++, i++;
else j = next[j];
if (j == m + 1)
{
ans[++tot] = i - m;
j = next[j];
}
}
}
int main()
{
scanf("%d%d%d", &n, &m, &s);
fo(i, 1, n)scanf("%d", &s1[i]);
fo(i, 1, m)scanf("%d", &s2[i]);
fo(i, 1, n)
{
fo(j, 1, s)rank1[i][j] = rank1[i - 1][j];
rank1[i][s1[i]]++;
}
fo(i, 1, m)
{
fo(j, 1, s)rank2[i][j] = rank2[i - 1][j];
rank2[i][s2[i]]++;
}
get();
kmp();
printf("%d\n", tot);
fo(i, 1, tot)printf("%d\n", ans[i]);
return 0;
}