HihoCoder1656 : 前缀后缀查询（[Offer收割]编程练习赛39）（字典树+小技巧）

描述

给定一个包含N个单词的字典:{W1, W2, W3, ... WN}，其中第i个单词Wi有具有一个权值Vi。

现在小Hi要进行M次查询，每次查询包含一个前缀字符串Pi和一个后缀字符串Si。他希望知道同时以Pi为前缀并且以Si为后缀的单词中，权值最大的单词的权值是多少？

假设字典包含"hihocoder"、"hijacker"和"hiker"，权值依次是30、20和10。

那么对于查询前缀="hi"，后缀="er"，答案应为30.

输入

第一行包两个整数N和M。(1 <= N <= M)

以下N行每行包含一个只包含小写字母的字符串Wi和对应的权值Vi。

再之后M行每行包含两个字符串Pi和Si。

对于30%的数据，1 <= N, M <= 100

对于100%的数据，1 <= N, M <= 50000, 1 <= |Wi|, |Pi|, |Si| <= 10， 1 <= Vi <= 100000

输出

输出最大的权值。如果没有符合条件的单词，输出-1。

样例输入

3 2
hihocoder 30
hijacker 20
hiker 10
hi er
hihoco hocoder

样例输出

30
30

初始代码：

思路是前缀查找，找到到Now节点，在Now的每个点（需要是单词尾）向前验证是否满足后缀要求，满足则更新最大值，无则输出-1。

这样显然是超时了，保证试一试的态度写了下。

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<memory>
using namespace std;
const int maxn=; char s[maxn];
int ch[maxn][],fa[maxn],id[maxn],num[maxn],cnt,val;
int max(int a,int b){ if(a>b) return a; return b; }
void insert()
{
    int Now=,L=strlen(s),x;
    for(int i=;i<L;i++){
        x=s[i]-'a';
        if(!ch[Now][x]) ch[Now][x]=++cnt,fa[cnt]=Now,id[cnt]=x;
        Now=ch[Now][x];
    }
    num[Now]=max(num[Now],val);
}
int Max; char a[],b[];
bool check(int Now)
{
    int L2=strlen(b);
    for(int i=L2-;i>=;i--){
        if(Now==) return false;
        if(b[i]-'a'!=id[Now]) return false;
        Now=fa[Now];
    } return true;
}
void dfs(int Now)
{
    if(num[Now]) if(check(Now)) Max=max(Max,num[Now]);
    for(int i=;i<;i++) {
       if(ch[Now][i])
            dfs(ch[Now][i]);
    }
}
void query()
{
    int Now=,L1=strlen(a),x;
    for(int i=;i<L1;i++){
        x=a[i]-'a';
        if(!ch[Now][x]) { printf("-1\n");return ;}
        Now=ch[Now][x];
    }
    Max=-; dfs(Now);
    printf("%d\n",Max);
}
int main()
{
    int n,m,i,L1,L2;
    scanf("%d%d",&n,&m);
    for(i=;i<=n;i++){
        scanf("%s%d",s,&val);
        insert();
    }
    for(i=;i<=m;i++){
        scanf("%s%s",a,b);
        L1=strlen(a);L2=strlen(b);
        query();
    }
    return ;
}

改进：

既然对象是前缀和后缀，显然不需要后缀数组或者后缀自动机（二者对象是任意子串）来完成。还是考虑字典树，想个办法把后缀前缀一起查询了。

具体：

对单词abcd （权值为Val）： 它的后缀是d，cd，bcd，abcd。把它的后缀连接到前缀上，成为4个单词（其他的最多也才10个），用'#'连接（后缀数组唱用）

对应     ：   d#abcd,cd#abcd,bcd#abcd,abcd#abcd，权值都为Val。

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<memory>
using namespace std;
const int maxn=; char s[maxn];
int ch[maxn<<][],num[maxn<<],cnt,Val;
int max(int a,int b){ if(a>b) return a; return b; }
void insert()
{
    int L=strlen(s),x;
    for(int j=L-;j>=;j--){
       int Now=;
       for(int i=j;i<L;i++){
          x=s[i]-'a';
          if(!ch[Now][x]) ch[Now][x]=++cnt;
          Now=ch[Now][x];
       }
       x=;
       if(!ch[Now][x]) ch[Now][x]=++cnt;
       Now=ch[Now][x];
       for(int i=;i<L;i++){
          x=s[i]-'a';
          if(!ch[Now][x]) ch[Now][x]=++cnt;
          Now=ch[Now][x];
          num[Now]=max(num[Now],Val);
       }
    }
}
char a[],b[];
void query()
{
    int Now=,L1=strlen(a),L2=strlen(b),x;
    for(int i=;i<L2;i++){
        x=b[i]-'a';
        if(!ch[Now][x]) { printf("-1\n");return ;}
        Now=ch[Now][x];
    }
    Now=ch[Now][];if(!Now) { printf("-1\n");return ;}
    for(int i=;i<L1;i++){
        x=a[i]-'a';
        if(!ch[Now][x]) { printf("-1\n");return ;}
        Now=ch[Now][x];
    }
    printf("%d\n",num[Now]);
}
int main()
{
    int n,m,L1,L2;
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++){
        scanf("%s%d",s,&Val);
        insert();
    }
    for(int i=;i<=m;i++){
        scanf("%s%s",a,b);
        query();
    }
    return ;
}