算法竞赛进阶指南：0x18：电话列表

给出一个电话列表，如果列表中存在其中一个号码是另一个号码的前缀这一情况，那么就称这个电话列表是不兼容的。

假设电话列表如下：

• Emergency 911
• Alice 97 625 999
• Bob 91 12 54 26

在此例中，报警电话号码（911）为 Bob 电话号码（91 12 54 26）的前缀，所以该列表不兼容。

输入格式

第一行输入整数 t，表示测试用例数量。

对于每个测试用例，第一行输入整数 n，表示电话号码数量。

接下来 n 行，每行输入一个电话号码，号码内数字之间无空格，电话号码不超过 10 位。

输出格式

对于每个测试用例，如果电话列表兼容，则输出 YES。

否则，输出 NO。

数据范围

1≤t≤40,
1≤n≤10000

输入样例：

2
3
911
97625999
91125426
5
113
12340
123440
12345
98346

输出样例：

NO
YES

思路：trie树

判断电话号码是否兼容可通过两个点：

1：当前串是否是某个串的前缀，判断当前串是否有新的节点，如果有，则是前缀，否则不是

2：是否存在某个串是当前串的前缀，用一个结束数组记录每个串结尾位置就可以了

 

#include<cstring>
#include<iostream>
using namespace std;

const int N=100010;

int son[N][10],idx;//trie模板
char str[20];
bool st[N];//记录每个串结束位置

bool add(char s[])
{
    bool new_node=false,prefix=false;
    //new_node表示当前串是否是其他串前缀，perfix表示其他串是否为当前串前缀
    int p=0;
    for(int i=0;s[i];++i)//模板
    {
        int u=s[i]-'0';
        if(!son[p][u])
        {
            son[p][u]=++idx;
            new_node=true;//当前串不是其他串前缀
        }
        p=son[p][u];
        if(st[p]) prefix=true;//其他串是当前串前缀
    }
    st[p]=true;
    
    return new_node&&!prefix;
}

int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        idx=0;//初始化
        memset(son,0,sizeof son);
        memset(st,false,sizeof st);
        bool res=true;
        
        int n;
        cin>>n;
        
        while(n--)
        {
            cin>>str;
            if(!add(str)) {
                res=false;
                
            }
        }
        if(res) puts("YES");
        else puts("NO");
    }
    return 0;
}