POJ 3167 Cow Patterns（模式串浮动匹配）

题目链接：http://poj.org/problem?id=3167

题意：模式串可以浮动的模式匹配问题给出模式串的相对大小，需要找出模式串匹配次数和位置。

思路：统计比当前数小，和于当前数相等的，然后进行kmp。

比如说模式串：1，4，4，2，3，1 而主串：5,6,2,10,10,7,3,2,9，那么2,10,10,7,3,2就是匹配的

code：

 #include <cstdio>
 #include <cstring>
 #include <vector>
 using namespace std;
 const int MAXN = ;
 const int MAXM = ;

 int a[MAXN];        // 存放主串
 int b[MAXM];        // 存放模式串
 int as[MAXN][];   // as[i][j] = k表示0 - i位中有k个数字j
 int bs[MAXM][];   // bs[i][j] = k表示0 - i位中有k个数字j
 int next[MAXM];     // 存放模式串失配时的移动位数
 vector<int> ans;    // 存放结果
 int n, m, s;

 void Init()
 {
     ans.clear();
     memset(as, , sizeof(as));
     memset(bs, , sizeof(bs));
     as[][a[]] = ;
     bs[][b[]] = ;
     for (int i = ; i <= n; ++i)
     {
         memcpy(as[i], as[i - ], sizeof(as[]));
         ++as[i][a[i]];
     }
     for (int i = ; i <= m; ++i)
     {
         memcpy(bs[i], bs[i - ], sizeof(bs[]));
         ++bs[i][b[i]];
     }
 }

 void GetNext()
 {
     memset(next, , sizeof(next));
     int i = , j = , k = ;
     next[] = ;
     while (i <= m)
     {
         int si = , sj = , ei = , ej = ;
         for (k = ; k < b[i]; ++k)
             si += bs[i][k] - bs[i - j][k];
         ei = bs[i][k] - bs[i - j][k];
         for (k = ; k < b[j]; ++k)
             sj += bs[j][k];
         ej = bs[j][k];
         if ( == j || (si == sj && ei == ej)) next[++i] = ++j;
         else j = next[j];
     }
 }

 void Kmp()
 {
     int i = , j = , k = ;
     while (i <= n)
     {
         int si = , sj = , ei = , ej = ;
         for (k = ; k < a[i]; ++k)
             si += as[i][k] - as[i - j][k];
         ei = as[i][k] - as[i - j][k];
         for (k = ; k < b[j]; ++k)
             sj += bs[j][k];
         ej = bs[j][k];
         if ( == j || (si == sj && ei == ej)) ++i, ++j;
         else j = next[j];
         if (j == m + )
         {
             ans.push_back(i - m);
             j = next[j];
         }
     }
 }

 int main()
 {
     while (scanf("%d %d %d", &n, &m, &s) == )
     {
         for (int i = ; i <= n; ++i) scanf("%d", &a[i]);
         for (int i = ; i <= m; ++i) scanf("%d", &b[i]);
         Init();
         GetNext();
         Kmp();
         size_t len = ans.size();
         printf("%d\n", len);
         for (size_t i = ; i < len; ++i) printf("%d\n", ans[i]);
     }
     return ;
 }