KMP算法 Trie树（9/9）

KMP算法

   int n, m;
  cin >> n >> a + 1 >> m >> b + 1;
这两行代码的意思是输入字符串a，b但是是从下标1开始输入的

 可以按照此方式输入字符串，但是输出必须按照下标for循环输出，不能直接输出

//注意next数组是对于要匹配的串写的（短的串）
 1 #include<iostream>
using namespace std;
const int N = 100010;
const int M = 1000010;
char a[N], b[M];
int ne[N];
int main()
{
    int n, m;
    cin >> n >> a + 1 >> m >> b + 1;

//给next数组进行编号
    for (int i = 2, j = 0; i <= n; i++)
    {
        while (j && a[i] != a[j + 1])j = ne[j];
        if (a[i] == a[j + 1])j++;
        ne[i] = j;
    }
//进行判断，验证利用next数组遍历
    for (int i = 1, j = 0; i <= m; i++)
    {
        while (j && b[i] != a[j + 1])j = ne[j];
        if (b[i] == a[j + 1])j++;
        if (j == n)
        {
            printf("%d ", i - n);
            j = ne[j];
        }
    }
    return 0;
}

 

Trie树

主要用于高效的存储和查找字符串

#include<iostream>
using namespace std;
const int N = 100010;
int a[N][26]; int idx;
int num[N];
void insert(char str[])
{
    int p = 0;//一开始的层数
    for (int i = 0; str[i]; i++)
    {
        int x = str[i] - 'a';
        if (!a[p][x]) a[p][x] = ++idx;如果没有就创建，但是存的位置赋予独一无二的值idx，和数组模拟链表类似
        p = a[p][x];//p走到下一个节点
    }
    num[p]++;//因为p最终到达地方唯一所以存入
}

int find(char str[])
{
    int p = 0;
    for (int i = 0; str[i]; i++)
    {
        int x = str[i] - 'a';
        if (!a[p][x]) return 0;
        p = a[p][x];
    }
    return num[p];//最后返回个数
}

int main()
{
    int n; cin >> n;
    while (n--)
    {
        char c; cin >> c; char str[N];
        if (c == 'I')
        {
            cin >> str;
            insert(str);
        }
        else {
            cin >> str;
            cout << find(str) << endl;
        }
    }
    return 0;
}