【数据结构OJ】A : DS串应用--KMP算法

Description

学习KMP算法，给出主串和模式串，求模式串在主串的位置

算法框架如下，仅供参考

#include<iostream>
#include<string>
using namespace std;
class myString
{
private:
    string mainstr; // 串
    int size;       // 串长度
    void GetNext(string p, int next[]);
    int KMPFind(string p, int pos, int next[]);
public:
    myString();
    ~myString();
    void SetVal(string sp);
    int KMPFindSubstr(string p, int pos);
};
myString::myString()
{
    size = 0;
    mainstr = "";
}
myString::~myString()
{
    size = 0;
    mainstr = "";
}
void myString::SetVal(string sp)
{
    mainstr = "";
    mainstr.assign(sp);
    size = mainstr.length();
}
int myString::KMPFindSubstr(string p, int pos)
{
    int i;
    int L = p.length();
    int *next = new int[L];
    GetNext(p, next);
    for(i = 0; i < L; i ++)
        cout << next[i] << ' ';
    cout << endl;
    int v = -1;
    v = KMPFind(p, pos, next);
    delete []next;
    return v;
}

Input

第一个输入t，表示有t个实例

第二行输入第1个实例的主串，第三行输入第1个实例的模式串

以此类推

Output

第一行输出第1个实例的模式串的next值

第二行输出第1个实例的匹配位置，位置从1开始计算，如果匹配成功输出位置，匹配失败输出0

以此类推

Sample

samplein

3
qwertyuiop
tyu
aabbccdd
ccc
aaaabababac
abac

sampleout

-1 0 0 
5
-1 0 1 
0
-1 0 0 1 
8

分析

基本代码都是差不多的，往上拼就行

AC代码

#include<iostream>
#include<string>
using namespace std;
class myString
{
private:
    string mainstr; // 串
    int size;       // 串长度
    void GetNext(string p, int next[]);
    int KMPFind(string p, int pos, int next[]);
public:
    myString();
    ~myString();
    void SetVal(string sp);
    int KMPFindSubstr(string p, int pos);
};
myString::myString()
{
    size = 0;
    mainstr = "";
}
myString::~myString()
{
    size = 0;
    mainstr = "";
}
void myString::SetVal(string sp)
{
    mainstr = "";
    mainstr.assign(sp);
    size = mainstr.length();
}
int myString::KMPFindSubstr(string p, int pos)
{
    int i;
    int L = p.length();
    int* next = new int[L+1];//这里对题目给的代码做了修改，确保有足够空间
    GetNext(p, next);
    for (i = 0; i < L; i++)
        cout << next[i] << ' ';
    cout << endl;
    int v = -1;
    v = KMPFind(p, pos, next);
    delete[]next;
    return v;
}

int myString::KMPFind(string p, int pos, int next[]) {
    int i=0, j=0;//i在主串循环，j在子串循环；
    int plen = p.length();
    int mlen = mainstr.length();
    while (i < mlen && j < plen) {
        if ((j == -1) || (mainstr[i] == p[j])) {
            ++i; ++j;;
        }
        else
            j = next[j];
    }
    if (j >= plen) {
        return i - p.length() + 1;//返回当前主串位置；
    }
    else
        return 0;
}

void myString::GetNext(string p, int next[]) {
    next[0] = -1;
    int j = -1, k = 0;
    while (k < p.length()) {
        if (j == -1 || p[k] == p[j]) {// P[k]表示后缀的单个字符，P[j]表示前缀的单个字符
            ++k; ++j; next[k] = j;
        }
        else {
            j = next[j];
        }
    }
}

int main() {
    int t;
    cin >> t;
    while (t--)
    {
        myString str;
        string mainstr, substr;
        cin >> mainstr >> substr;
        str.SetVal(mainstr);
        int ans;
        ans = str.KMPFindSubstr(substr, 0);
        cout << ans << endl;

    }
    return 0;
}