【KMP】【poj3167】【bzoj1729】Cow Patterns (kmp变形)

最新推荐文章于 2018-09-14 18:52:16 发布

原创最新推荐文章于 2018-09-14 18:52:16 发布 · 434 阅读

0 ·

CC 4.0 BY-SA版权

----kmp 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了KMP算法，并结合poj3167和bzoj1729两个编程题，探讨了KMP算法在字符串匹配问题中的变形应用。

两个序列的偏序关系相同当且仅当这两个序列的每个位置上的元素前面等于它的元素个数和小于它的元素个数都相等

题意
给定一个顺序序列表示奶牛的排列顺序，定义串a=b当且仅当a的每位的排名=b的每位的排名。

参考:http://blog.youkuaiyun.com/waitfor_/article/details/7717059

    #include<stdio.h>  
    #include<string.h>  
    int a[100010],b[25010];  
    int c[30];                //树状数组中的元素  
    int m1[25010],m2[25010];  //（匹配串中）m1表示小于，m2表示小于等于  
    int n,k,s;  
    int next[25010];  
    int pop[100010],all;  
    int lowbit(int x){  
        return x&(x^(x-1));  
    }  
    int sum(int x){  
        int ss=0;  
        while(x>0){  
            ss+=c[x];  
            x-=lowbit(x);  
        }  
        return ss;  
    }  
    void update(int x,int val){  
        while(x<=s){  
            c[x]+=val;  
            x+=lowbit(x);  
        }  
    }  
    void getnext(){  
        int i=1,j=0,kk;  
        next[1]=0;  
        while(i<=k){  
            if(j==0 || (sum(b[i]-1)==m1[j] && sum(b[i])==m2[j])){  
                i++,j++;  
                next[i]=j;  
                if(i==k+1) return ;  
                update(b[i],1);  
            }  
            else{  
                for(kk=i-j+1;kk<=i-next[j];kk++)        //这里对next[j]之前的b[i]都要消除  
                    update(b[kk],-1);  
                j=next[j];  
            }  
        }  
    }   
    int kmp(){                //这里对待匹配串用树状数组记录  
        int i=1,j=1,f;  
        update(a[1],1);  
        while(i<=n){  
            if(j==0 ||(sum(a[i]-1)==m1[j] && sum(a[i])==m2[j])){  
                ++i,++j;  
                if(i<=n)  
                    update(a[i],1);  
            }  
            else{  
                for(f=i-j+1;f<=i-next[j];f++)  
                    update(a[f],-1);  
                j=next[j];  
            }  
            if(j>k){  
                for(f=i-j+1;f<=i-next[j];f++)  
                    update(a[f],-1);  
                all++;  
                pop[all]=i-1;  
                j=next[j];  
            }  
        }  
        return all;  
    }  
    int main(){  
        int i,j;  
        scanf("%d %d %d",&n,&k,&s);  
        for(i=1;i<=n;i++)  
            scanf("%d",&a[i]);  

        memset(c,0,sizeof(c));  
        memset(m1,0,sizeof(m1));  
        memset(m2,0,sizeof(m2));  

        for(i=1;i<=k;i++){  
            scanf("%d",&b[i]);  
            update(b[i],1);  
            m1[i]=sum(b[i]-1);  
            m2[i]=sum(b[i]);  
        }  

        memset(c,0,sizeof(c));  
        getnext();  

        memset(c,0,sizeof(c));  
        printf("%d\n",kmp());  
        for(i=1;i<=all;i++)  
            printf("%d\n",pop[i]+1-k);  
    }