zoj 3643 Keep Deleting

思路:KMP+堆栈,很巧妙的思路,用栈记录下每一次匹配成功时候的位置,当完全匹配成功的时候,就弹出相应程度的堆栈内容,原来一直错的地方在于,每次放到堆栈里的是陪陪失败后的位置,但是那样的话会有重复的情况,所以应该果断换一种算法。


/*#########################################################################
# File Name: k.cpp
# Author: CaoLei
# Created Time: 2015/7/14 16:11:49
#########################################################################*/

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <set>
#include <stack>
#include <queue>
#include <map>
using namespace std;
#define MAX(x,y) (((x)>(y))?(x):(y))
#define MIN(x,y) (((x)<(y))?(x):(y))
#define N 500010
#define pi acos(-1.0)
#define inf 100000000
typedef long long ll;
typedef unsigned long long ull;
char a[300],b[550000];
int fnext[300],n,m;
void build(){
    int i,j=0;
    fnext[1]=0;
    for(i=2;i<=n;i++){
        while(j>0&&a[j+1]!=a[i]) j=fnext[j];
        if(a[j+1]==a[i]) j++;
        fnext[i]=j;
    }
}
int kmp(){
    int ans=0;
    stack<int> s;
    int j=0,tmp;
    for(int i=1;i<=m;i++){
        while(j>0&&a[j+1]!=b[i]){
            j=fnext[j];
        }
        if(a[j+1]==b[i]) j++;
        s.push(j);
        if(j==n){
            for(int cnt=1;cnt<=n;cnt++){
                s.pop();
            }

            if(s.empty()) j=0;
            else {
                    j=s.top();
                }
            ans++;
        }
    }
    if(!s.empty()) s.pop();
    return ans;
}

int main(){
    //freopen("in.txt","r",stdin);
    while(~scanf("%s%s",a+1,b+1)){
        n=strlen(a+1);
        m=strlen(b+1);
        memset(fnext,0,sizeof(fnext));
        build();
        printf("%d\n",kmp());
    }
    return 0;
}




你可能感兴趣的:(zoj 3643 Keep Deleting)