[CodingHarder] Trie树

https://vjudge.net/contest/311647#overview

A - Shortest Prefixes

给定n个串，求出每个串的“唯一最短前缀”

用bo[i]表示从字典树根节点0到节点i连成的字符串为前缀的单词个数，那么在查询一个单词的时候，在该路径的某个节点i出现了bo[i]=1的情况，就说明这个字符串是该单词的唯一前缀。

#include<iostream>
#include<sstream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<iomanip>
#include<cstdlib>
#include<cctype>
#include<vector>
#include<string>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<map>
#include<set>
#define mem(a,b) memset(a,b,sizeof(a))
#define random(a,b) (rand()%(b-a+1)+a)
#define ll long long
#define ull unsigned long long
#define e 2.71828182
#define Pi acos(-1.0)
#define ls(rt) (rt<<1)
#define rs(rt) (rt<<1|1)
#define lowbit(x) (x&(-x))
using namespace std;
const int MAXN=2e4+5;
const int Z=26;
int ch[MAXN][Z];
int bo[MAXN];
char str[1001][21];
int tot=1;
void insert(char *s)
{
    int l=strlen(s),u=1;
    for(int i=0;i<l;++i)
    {    
        int c=s[i]-'a';
        if(!ch[u][c]) ch[u][c]=++tot;
        u=ch[u][c];
        bo[u]++;
    }
}
void query(char *s)
{
    int l=strlen(s),u=1;
    for(int i=0;i<l;++i)
    {
        int c=s[i]-'a';
        if(bo[u]==1) break;
        putchar(s[i]);
        u=ch[u][c];
    }
    putchar('
');
}
int main()
{
    int cnt=0;
    while(scanf("%s",str[++cnt])!=EOF)
    insert(str[cnt]);
    
    for(int i=1;i<=cnt;++i)
    {
        cout<<str[i]<<' ';
        query(str[i]);
    }
}

B - Wild Words

 模糊匹配，一个'?'字符匹配任意一个字符，一个'*'字符匹配0个、1个或多个字符。

dfs和Trie的结合

两个指针在查询的字符串和Trie上走

#include<iostream>
#include<sstream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<iomanip>
#include<cstdlib>
#include<cctype>
#include<vector>
#include<string>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<map>
#include<set>
#define mem(a,b) memset(a,b,sizeof(a))
#define random(a,b) (rand()%(b-a+1)+a)
#define ll long long
#define ull unsigned long long
#define e 2.71828182
#define Pi acos(-1.0)
#define ls(rt) (rt<<1)
#define rs(rt) (rt<<1|1)
#define lowbit(x) (x&(-x))
using namespace std;
const int MAXN=6e5+5;
//const int MAXN=1e5+5;
const int Z=28;
int ch[MAXN][Z];
vector<int> bo[MAXN];
vector<int> ans;
char str[30];
bool isok[MAXN];
bool visit[MAXN];
int n,m,len;
int tot=1;
int read()
{
    int s=1,x=0;
    char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') s=-1;ch=getchar();}
    while(isdigit(ch)) {x=10*x+ch-'0';ch=getchar();}
    return x*s;
}
int f(char c)
{
    if(c=='?') return 26;
    if(c=='*') return 27;
    return c-'a';
}
void insert(char *s,int id)
{
    int l=strlen(s),u=1;
    for(int i=0;i<l;++i)
    {
        if(!ch[u][f(s[i])]) ch[u][f(s[i])]=++tot;
        u=ch[u][f(s[i])];
        if(f(s[i])==27) isok[u]=true;
    }
    bo[u].push_back(id);
}
void find(int le,int u)
{
    if(le==len)
    {
        int sz=bo[u].size();
        for(int i=0;i<sz;++i)
        if(!visit[bo[u][i]]) 
        {
            visit[bo[u][i]]=true;
            ans.push_back(bo[u][i]);
        }
        if(ch[u][27]) find(le,ch[u][27]);
        return;
    }
    int c=f(str[le]);
    if(ch[u][c]) find(le+1,ch[u][c]);//精准匹配 
    if(ch[u][26]) find(le+1,ch[u][26]);//?字符匹配 
    if(ch[u][27])//*字符匹配 
    {
        find(le+1,ch[u][27]);//匹配1个 
        find(le,ch[u][27]);//匹配0个 
    }
    if(isok[u]) find(le+1,u);//*匹配多个 
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin>>n>>m;
    for(int i=1;i<=n;++i)
    {
        cin>>str;
        insert(str,i-1);
    }
    for(int i=1;i<=m;++i)
    {
        cin>>str;
        len=strlen(str);
        ans.clear();
        mem(visit,0);
        find(0,1);
        int sz=ans.size();
        if(!sz)
        {
            cout<<"Not match"<<endl;
            continue;
        }
        sort(ans.begin(),ans.end());
        for(int j=0;j<sz;++j)
        {
            if(j==sz-1) cout<<ans[j]<<endl;
            else cout<<ans[j]<<' ';
        }
    }
    return 0;
}

C - Phone List

给定n个长度不超过10的数字串，问其中是否存在两个数字串S、T，使得S是T的前缀。
将所有的字符串构成一棵Trie，在构建过程中可以顺便判断答案。
若当前串插入后没有新建任何结点，则当前串肯定是之前插入的某个串的前缀；
若插入过程中，有某个经过的结点带有串结尾标记，则之前插入的某个串是当前串的前缀 

#include<iostream>
#include<sstream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<iomanip>
#include<cstdlib>
#include<cctype>
#include<vector>
#include<string>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<map>
#include<set>
#define mem(a,b) memset(a,b,sizeof(a))
#define random(a,b) (rand()%(b-a+1)+a)
#define ll long long
#define ull unsigned long long
#define e 2.71828182
#define Pi acos(-1.0)
#define ls(rt) (rt<<1)
#define rs(rt) (rt<<1|1)
#define lowbit(x) (x&(-x))
using namespace std;
const int MAXN=2e5+5;
const int Z=10;
int ch[MAXN][Z];
int tot=1;
bool bo[MAXN];
bool insert(char *s)
{
    int l=strlen(s),u=1;
    bool flag=false;
    for(int i=0;i<l;++i)
    {
        int c=s[i]-'0';
        if(!ch[u][c]) ch[u][c]=++tot;
        else if(i==l-1) flag=true;
        u=ch[u][c];
        if(bo[u]) flag=true;
     }
     bo[u]=true;
     return flag;
}
int read()
{
    int s=1,x=0;
    char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') s=-1;ch=getchar();}
    while(isdigit(ch)) {x=10*x+ch-'0';ch=getchar();}
    return x*s;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    int test;cin>>test;
    while(test--)
    {
        int n;cin>>n;
        bool flag=false;
        mem(ch,0);mem(bo,0);
        tot=1;
        char str[20];
        for(int i=1;i<=n;++i)
        {
            cin>>str;
            if(insert(str)) flag=true;
        }
        if(flag) cout<<"NO"<<endl;
        else cout<<"YES"<<endl;
    }
}