hdu 1671 Phone List Trie树

题意：给定一个字符串的集合，看是否存在一个字符串是另一个字符串的前缀

思路：直接上lrj的模板， 话说深感此模板强大，比动态分配简洁许多，就是耗内存多，细节看代码

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1671

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

int n;
char s[20];
bool flag;

struct Trie
{
    int ch[100000][15];
    int val[100000];
    int sz;

    int id(char c)
    {
        return c - '0';
    }

    void init()
    {
        sz = 1;
        memset(ch[0], 0, sizeof(ch[0]));
        memset(val, 0,sizeof(val));
    }

    void Insert(const char *s)
    {
        int u = 0, len = strlen(s);
        for(int i=0; i<len; i++)
        {
            int c = id(s[i]);
            if(!ch[u][c])
            {
                memset(ch[sz], 0, sizeof(ch[sz]));
                ch[u][c] = sz++;
                val[u] = 0;
            }
            else
            {
                if(val[ch[u][c]] == 1) //之前的字符串是当前字符串的前缀
                {
                    flag = false;
                    return ;
                }
            }
            u = ch[u][c];
        }
        val[u] = 1;
        for(int i=0; i<10; i++) //当前字符串是之前的字符串的前缀
            if(ch[u][i]) flag = false;
    }
};

Trie trie;

int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        trie.init();
        flag = true;
        for(int i=0; i<n; i++)
        {
            scanf("%s", s);
            if(flag)
                trie.Insert(s);
        }
        if(flag)
            puts("YES");
        else
            puts("NO");
    }
    return 0;
}