POJ 3167 Cow Patterns KMP+暴力

来源：互联网发布：centos macaddr 不写编辑：程序博客网时间：2024/05/20 07:36

时空隧道

题意：
给出两个串，如果a和b匹配当且仅当ab中的每个对应位置的数字rank1和rank2相同，rank1代表当前元素前面小于他的元素个数，rank2代表当前元素前面等于他的元素个数，求出匹配串和模式串的匹配位置（开头位置）

分析：
KMP…裸的KMP…就是计算一下rank就好…
因为数字不会超过25，所以直接暴力求rank就好…正解其实是树状数组…

代码如下：

#include<algorithm>#include<iostream>#include<cstring>#include<cstdio>//by NeighThornusing namespace std;const int maxn=100000+5,maxk=25000+5,maxs=25+5;int n,k,s,a[2][maxn],nxt[maxk],num[2][maxn][maxs],vis[maxk];inline int rank1(int len,int en,int id){    int st=en-len,ans=0;    for(int i=1;i<a[id][en];i++)        ans+=num[id][en][i]-num[id][st][i];    return ans;}inline int rank2(int len,int en,int id){    return num[id][en][a[id][en]]-num[id][en-len][a[id][en]];}inline void getnext(void){    nxt[0]=nxt[1]=0;int p;    for(int i=1;i<k;i++){        p=nxt[i];int a=rank1(p+1,p+1,1),b=rank1(p+1,i+1,1),c=rank2(p+1,p+1,1),d=rank2(p+1,i+1,1);        while(p&&(a!=b||c!=d))            p=nxt[p],a=rank1(p+1,p+1,1),b=rank1(p+1,i+1,1),c=rank2(p+1,p+1,1),d=rank2(p+1,i+1,1);        if(rank1(p+1,p+1,1)==rank1(p+1,i+1,1)&&rank2(p+1,p+1,1)==rank2(p+1,i+1,1))            nxt[i+1]=p+1;        else            nxt[i+1]=0;    }}inline void kmp(void){    int posa=1,posb=1,ans=0,stk[maxn];    while(posa<=n){        if(rank1(posb,posa,0)==rank1(posb,posb,1)&&rank2(posb,posa,0)==rank2(posb,posb,1))            posa++,posb++;        else if(posb==1)            posa++;        else            posb=nxt[posb-1]+1;        if(posb==k+1)            stk[++ans]=posa-k,posb=nxt[posb-1]+1;    }    cout<<ans<<endl;    for(int i=1;i<=ans;i++)        cout<<stk[i]<<endl;}signed main(void){    scanf("%d%d%d",&n,&k,&s);memset(num,0,sizeof(num));    for(int i=1;i<=n;i++)        scanf("%d",&a[0][i]),num[0][i][a[0][i]]=1;    memset(vis,0,sizeof(vis));    for(int i=1;i<=n;i++)        for(int j=1;j<=s;j++)            num[0][i][j]+=num[0][i-1][j];    for(int i=1;i<=k;i++)        scanf("%d",&a[1][i]),num[1][i][a[1][i]]=1;    for(int i=1;i<=k;i++)        for(int j=1;j<=s;j++)            num[1][i][j]+=num[1][i-1][j];    getnext();kmp();    return 0;}

by >_< NeighThorn

1 0