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

来源:互联网 发布:淘宝店铺没人买 编辑:程序博客网 时间:2024/06/06 00:41

题意:给定一个顺序序列表示奶牛的排列顺序,定义串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[N], s2[N];int rank1[N][30], rank2[N][30];int next[N], ans[N];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;}
阅读全文
0 0
原创粉丝点击