POJ - 3461(Oulipo)

題目大意

給定兩個字符串 S 和 P ，問 P 在 S 中出現的次數。

解題思路

字符串匹配算法之KMP模板題。點擊下方鏈接查看KMP算法模板
字符串匹配之KMP(輔助理解KMP算法模板)

#include <iostream>
#include <string>
using namespace std;
const int MAXN = 10005;
int ne[MAXN];   //next數組,ne[j]表示P~0...j~字符串的前綴與後綴相等的最長長度-1（-1是因爲下標從0開始，實質上是最大長度）
void init(int n, string p) {  //n表示模板串長度,p表示模板串
    ne[0] = -1;     //-1表示長度爲0
    for(int i = 1, j = -1; i < n; i++) {   //i指向文本串,j指向模式串 
        while(j != -1 && p[i] != p[j+1]) {      
            j = ne[j];
        }
        if(p[i] == p[j+1]) {
            j++;
        }
        ne[i] = j;      
    }
}
int main() {
    int T;
    cin >> T;
    while(T--) {
        string s, p;    //s表示文本串,p表示模式串
        cin >> p >> s;
        int n = p.length(), m = s.length();
        init(n, p);
        int res = 0;
        for(int i = 0, j = -1; i < m; i++) {    //m表示文本串長度
            while(j != -1 && s[i] != p[j+1]) {  //不匹配則回溯j到ne[j]位置
                j = ne[j];
            }
            if(s[i] == p[j+1]) {    //當前位置匹配,判斷下一個位置
                j++;
            }
            if(j == n-1) {      //匹配到結尾,則再文本串位置找到模式串
                res++;          //res表示模式串在匹配串中個數,在求其他問題時,可變換
                j = ne[j];      //繼續尋找下一個模式串
            }
        }
        cout << res << endl;
    }
    return 0;
}