hdu 1671 字典树

最新推荐文章于 2021-07-23 14:32:50 发布

yo_haha

最新推荐文章于 2021-07-23 14:32:50 发布

阅读量279

点赞数

分类专栏：字典树 HDOJ

本文链接：https://blog.csdn.net/yo_haha/article/details/46419949

版权

HDOJ 同时被 2 个专栏收录

21 篇文章 0 订阅

订阅专栏

字典树

2 篇文章 0 订阅

订阅专栏

http://acm.hdu.edu.cn/showproblem.php?pid=1671
简单的字典树注意每次要释放内存不然会MLE

#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
using namespace std;
struct node
{
    int sign;
    node *next[10];
    node()
    {
        memset(next,NULL,sizeof(next));
        sign=0;
    }
};
struct number
{
    string num;
}a[10001];
bool cmp(number a, number b)
{
    return a.num.size()<b.num.size();
}
node *root = new node();
int insert(string s)
{
    node *p = root;
    int len = s.size();
    for(int i = 0; i<len; i++)
    {
        int k = s[i]-'0';
        if(p->next[k]==NULL)
            p->next[k]=new node();
        if(p->sign)
            return 1;
        p=p->next[k];
    }
    p->sign=1;
    return 0;
}
void del(node *p)
{
    for(int i = 0; i<10; i++)
        if(p->next[i]!=NULL)
            del(p->next[i]);
    delete p;
}
int main()
{
    int t,n;
    cin>>t;
    while(t--)
    {
        root = new node();
        int flag=1;
        cin>>n;
        for(int i = 0; i<n; i++)
            cin>>a[i].num;
        sort(a,a+n,cmp);
        for(int i = 0; i<n&&flag; i++)
            if(insert(a[i].num))
                flag=0;
        if(flag)
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
        del(root);
    }
}