BZOJ1212 [HNOI2004]L语言

看到最长前缀想到Trie字典树

但此题是分段匹配（多个单词），于是想到二重循环枚举起点

记录\(mark_i\)表示能否通过单词匹配到\(i\)，如果可以则记录答案并以该点为起点继续匹配

若匹配过程中发现之前单词的结尾（可用结尾标记判断），则令当前\(mark=1\)

注：\(1MB\approx10^6\)

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

bool bo[1000005];//1MB=1e6
int ch[1000005][30],n,m,tot = 1;
char str[1000005];
bool mark[1000005];//表示能通过多个单词匹配到i位置 
inline void write(int s)
{
    if(s < 0)
    {
        putchar('-');
        s = -s;
        write(s);
    }
    if(s > 9)
        write(s % 10);
    putchar(s % 10 + '0');
}
inline void Insert(char * s)//建Trie字典树 
{
    int u = 1,len = strlen(s + 1);
    for(int i = 1;i <= len; ++i)
    {
        int c = s[i] - 'a';
        if(!ch[u][c])
            ch[u][c] = ++tot;
        u = ch[u][c];
    }
    bo[u] = true;
}
inline void find(char * s)//匹配文章s 
{
    int ans,now = 1,len = strlen(s + 1);
    memset(mark,0,sizeof mark);
    mark[0] = 1;//0位置匹配根节点 
    for(int i = 0,k;i <= len; ++i)//从0开始！ 
    {
        if(!mark[i])//不能匹配到当前位置 
            continue;
        else
            ans = i;//记录答案 
        for(k = i + 1,now = 1;k <= len; ++k)
        {
            int c = s[k] - 'a';
            now = ch[now][c];
            if(!now)
                break;
            if(bo[now])//发现单词结尾 
                mark[k] = 1;
        }
    }
    cout << ans << endl;
}

int main()
{
    ios::sync_with_stdio(false);
    cin >> n >> m;
    for(int i = 1;i <= n; ++i)
    {
        cin >> str + 1;
        Insert(str);
    }
    for(int i = 1;i <= m; ++i)
    {
        cin >> str + 1;
        find(str);
    }
    return 0;
}