CF #344 D. Messenger KMP/Z

题目链接：http://codeforces.com/problemset/problem/631/D

给定两个压缩形式的字符串，如a3b5a4k7这样的形式

问A在B中出现次数。

分类讨论，如果A是只有一种字符的，则答案数量可能很大，但计算也很简单，直接看B的每一个字符，答案累加上cnt2-cnt1+1

如果A不是单字符的，则答案至多是B的压缩之后长度的数量级。

不考虑A的第一个字符，用KMP或者Z函数来计算A的出现情况，如果匹配长度为lenA-1，则检查是否该匹配的第一个字符与A【0】相等且数量大于A【0】的数量，如果匹配长度为lenA-2，则检查是否下一个字符与A最后一个字符是相等的，且数量大于A【lenA-1】的数量。

写的时候需要仔细，因为涉及到不少下标以及条件判断。

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
#include <string.h>
#include <stdio.h>
#include <math.h>
#include <stdlib.h>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <ctime>
#include <numeric>
#include <cassert>

using namespace std;

const int N=1e6+100;


struct Char {
    char ch;
    long long cnt;
    bool operator == (const Char &o) const {
        return ch==o.ch&&cnt==o.cnt;
    }
};
int z[N];
Char f[N];
void Z(int n) {
    z[0]=n;
    int L=0,R=0;
    for (int i=1;i<n;i++) {
        if (i>R) {
            L=i,R=i;
            while (R<n&&f[R-i]==f[R]) R++;
            z[i]=R-L;
            R--;
        }
        else {
            int k=i-L;
            if (z[k]<R-i+1)
                z[i]=z[k];
            else {
                L=i;
                while (R<n&&f[R-i]==f[R]) R++;
                z[i]=R-L;
                R--;
            }
        }
    }
}
Char s[N],t[N];


int main() {
    int n,m;
    scanf("%d %d",&n,&m);
    char buf[5];
    for (int i=0;i<n;i++) {
        long long cnt;
        scanf("%I64d-%s",&cnt,buf);
        t[i].cnt=cnt;
        t[i].ch=buf[0];
    }
    int k=1;
    for (int i=1;i<n;i++) {
        if (t[i].ch==t[i-1].ch)
            t[k-1].cnt+=t[i].cnt;
        else {
            t[k++]=t[i];
        }
    }
    n=k;
    for (int i=0;i<m;i++) {
        long long cnt;
        scanf("%I64d-%s",&cnt,buf);
        s[i].cnt=cnt;
        s[i].ch=buf[0];
    }
    k=1;
    for (int i=1;i<m;i++) {
        if (s[i].ch==s[i-1].ch)
            s[k-1].cnt+=s[i].cnt;
        else {
            s[k++]=s[i];
        }
    }
    m=k;

    if (m==1) {
        long long ret=0;
        for (int i=0;i<n;i++) {
            if (t[i].ch==s[0].ch&&t[i].cnt>=s[0].cnt) {
                ret+=t[i].cnt-s[0].cnt+1;
            }
        }
        printf("%I64d
",ret);
        return 0;
    }
    int len=0;
    for (int i=1;i<m;i++) {
        f[len++]=s[i];
    }
    f[len].ch='#';
    f[len].cnt=0;
    len++;
    int from=len;
    for (int i=0;i<n;i++) {
        f[len++]=t[i];
    }
    Z(len);
    int ret=0;
    for (int i=from+1;i<len;i++) {
        int lcp=z[i];
        if (t[i-from-1].ch==s[0].ch&&t[i-from-1].cnt>=s[0].cnt) {
            if (lcp==m-1)
                ret++;
            else if (lcp==m-2) {
                if (s[m-1].ch==t[i-from+lcp].ch&&s[m-1].cnt<=t[i-from+lcp].cnt)
                    ret++;
            }
        }
    }
    printf("%d
",ret);
    return 0;
}