HDU 1671(字典树)

字母换成数字而已,字典树模板。

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
const int maxn=10005;
const double eps=1e-8;
const double PI = acos(-1.0);
#define lowbit(x) (x&(-x))
char s[maxn][11];
int len,trie[maxn*11][12],sum[maxn*11],root,tot;
void insert(char *s)
{
    len=strlen(s);
    root=0;
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'0';
        if(!trie[root][id]) trie[root][id]=++tot;
        sum[trie[root][id]]++;//前缀后移一个位置保存
        root=trie[root][id];
    }
}
int search(char *s)
{
    root=0;
    len=strlen(s);
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'0';
        if(!trie[root][id]) return 0;
        root=trie[root][id];
    }//root经过此循环后变成前缀最后一个字母所在位置的后一个位置
    return sum[root];//因为前缀后移了一个保存,所以此时的sum[root]就是要求的前缀出现的次数
}
int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);
    std::cout.tie(0);
    int t;
    cin>>t;
    while(t--)
    {
        tot=0;
        memset(trie,0,sizeof(trie));
        memset(sum,0,sizeof(sum));
        int n;
        cin>>n;
        for(int i=0;i<n;i++)
        {
            cin>>s[i];
            insert(s[i]);
        }
        int f=0;
        for(int i=0;i<n;i++)
        {
            if(search(s[i])>1)
            {
                cout<<"NO"<<endl;
                f=1;
                break;
            }
        }
        if(!f)
        {
            cout<<"YES"<<endl;
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Dilly__dally/article/details/81712527