洛谷P2292 [HNOI2004]L语言（trie+dp）

[HNOI2004]L语言

题目描述

标点符号的出现晚于文字的出现，所以以前的语言都是没有标点的。现在你要处理的就是一段没有标点的文章。

一段文章T是由若干小写字母构成。一个单词W也是由若干小写字母构成。一个字典D是若干个单词的集合。我们称一段文章T在某个字典D下是可以被理解的，是指如果文章T可以被分成若干部分，且每一个部分都是字典D中的单词。

例如字典D中包括单词{‘is’, ‘name’, ‘what’, ‘your’}，则文章‘whatisyourname’是在字典D下可以被理解的，因为它可以分成4个单词：‘what’, ‘is’, ‘your’, ‘name’，且每个单词都属于字典D，而文章‘whatisyouname’在字典D下不能被理解，但可以在字典D’=D+{‘you’}下被理解。这段文章的一个前缀‘whatis’，也可以在字典D下被理解，而且是在字典D下能够被理解的最长的前缀。

给定一个字典D，你的程序需要判断若干段文章在字典D下是否能够被理解。并给出其在字典D下能够被理解的最长前缀的位置。

输入输出格式

输入格式：
输入文件第一行是两个正整数n和m，表示字典D中有n个单词，且有m段文章需要被处理。之后的n行每行描述一个单词，再之后的m行每行描述一段文章。

其中1<=n, m<=20，每个单词长度不超过10，每段文章长度不超过1M。

输出格式：
对于输入的每一段文章，你需要输出这段文章在字典D可以被理解的最长前缀的位置。

输入输出样例

输入样例#1：
4 3
is
name
what
your
whatisyourname
whatisyouname
whaisyourname
输出样例#1：
14 （整段文章’whatisyourname’都能被理解）
6 （前缀’whatis’能够被理解）
0 （没有任何前缀能够被理解）

分析：对字典D建一棵trie，dp[i]表示文章前i位是否能被读懂，然后转移即可，答案显然。

代码

#include <cstdio>
#include <string>
#include <cstring>
#define N 10000
using namespace std;

struct node
{
    int nxt[26],en;
}trie[N];
int n,m,num,ans,f[N*100];
char s[N*100];
bool fl;

void ins(char *s)
{
    int len=strlen(s);
    int u=0;
    for (int i=0;i<len;i++)
    {
        int v=s[i]-'a';
        if (!trie[u].nxt[v]) trie[u].nxt[v]=++num;
        u=trie[u].nxt[v];
    }
    trie[u].en=1;
}

void dp(int k,char *s)
{
    int len=strlen(s);
    for (int i=-1;i<len;i++)
    {
        if (i>-1&&f[i]!=k) continue;
        ans=i+1;
        int u=0;
        for (int j=i+1;j<len;j++)
        {
            int v=s[j]-'a';
            u=trie[u].nxt[v];
            if (!u) break;
            if (trie[u].en) f[j]=k;
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
    {
        scanf("%s",s);
        ins(s);
    }
    for (int i=1;i<=m;i++)
    {
        scanf("%s",s);
        dp(i,s);
        printf("%d\n",ans);
    }
}