POJ 3167 Cow Patterns KMP+暴力

最新推荐文章于 2018-04-03 18:31:00 发布

NeighThorn

最新推荐文章于 2018-04-03 18:31:00 发布

阅读量515

点赞数 1

分类专栏： poj题解

本文链接：https://blog.csdn.net/neighthorn/article/details/52993677

版权

poj题解专栏收录该内容

94 篇文章 0 订阅

订阅专栏

时空隧道

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

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

代码如下：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
//by NeighThorn
using 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;
}