Codeforces 631D Messenger【KMP】

来源:互联网 发布:spss如何将数据分组 编辑:程序博客网 时间:2024/06/16 16:34

题意:

给定由字符串块(字符及连续出现的个数)组成的字符串t,s,求t串中有多少个s。

分析:

KMP
这题唯一需要思考的地方就是如何处理字符串块。第一想到是把他们都展开然后进行KMP,可是展开后实在太长,所以必须按块进行处理,就要把所有相邻的相同的块进行合并成一个大块。
注意模式串开头和结尾处与t中对应的块不需要严格相等,只要字符相同并且t中个数不小于模式串中的个数~~所以只需将模式串去掉开头和结尾进行匹配,最后再判断一下就好啦~
既然要去掉开头结尾,就要保证原模式串s的长度大于2,所以长度为1和2的情况都需要特殊考虑一下~

代码:

#include<cstdio>#include<cstring>#include<iostream>using namespace std;const int maxn = 200005;typedef long long ll;typedef pair<char, ll> pii;#define fi first#define se secondpii s[maxn], t[maxn], ms[maxn];int m, n, mm;int next[maxn];void KMP_pre(){    int j = next[0] = -1;    int i = 0;    while(i < mm){        while(j != -1 && ms[i] != ms[j])  j = next[j];        next[++i] = ++j;    }}ll KMP_count(){    int i = 0, j = 0;    ll ans = 0;    while(i < n){        while(j != -1 && t[i]!= ms[j]) j = next[j];        i++, j++;        if(j >= mm){            if(i < n && t[i - mm - 1].fi == s[0].fi && t[i - mm - 1].se >= s[0].se               && t[i].fi == s[m - 1].fi && t[i].se >= s[m - 1].se){                ans++;               }            j = next[j];        }    }    return ans;}int main (void){   int N, M;   scanf("%d%d", &N, &M);   m = n = mm = 0;   ll tl;   char tc;   for(int i = 0; i < N; i++){        scanf("%I64d-%c", &tl, &tc);        if(i == 0){            t[n++] = pii(tc, tl);        }else{            if(tc == t[n - 1].fi)                t[n - 1].se += tl;            else                t[n++] = pii(tc, tl);        }   }   //for(int i = 0; i < n; i++) cout<<t[i].fi<<' '<<t[i].se<<endl;    for(int i = 0; i < M; i++){        scanf("%I64d-%c", &tl, &tc);        if(i == 0){            s[m++] = pii(tc, tl);        }else{            if(tc == s[m - 1].fi)                s[m - 1].se += tl;            else                s[m++] = pii(tc, tl);        }   } // for(int i = 0; i < m; i++) cout<<s[i].fi<<' '<<s[i].se<<endl;   if(m == 1){        ll ans = 0;         for(int i = 0; i < n; i++){            if(t[i].fi== s[0].fi && t[i].se >= s[0].se)                ans += t[i].se - s[0].se + 1;         }        cout<<ans<<endl;   }else if(m == 2){       ll ans = 0;        for(int i = 0; i < n - 1; i++){            if(t[i].fi == s[0].fi && t[i].se >= s[0].se               && t[i +1].fi == s[1].fi && t[i + 1].se >= s[1].se)                 ans++;        }        cout<<ans<<endl;   }else{       mm = 0;        for(int i = 1; i < m - 1; i++)            ms[mm++] = s[i];        KMP_pre();        printf("%I64d\n",KMP_count());   }   return 0;}

思考思考思考~相信自己一定可以做出来的~~


吐槽:为啥么越来越觉得D比C 好做呢~

0 0
原创粉丝点击