POJ 3630 Phone List（trie树的简单应用）

题目链接：http://poj.org/problem?id=3630

题意：给你多个字符串，如果其中任意两个字符串满足一个是另一个的前缀，那么输出NO，否则输出YES

思路：简单的trie树应用，插入的过程中维护到当前节点是不是字符串这个布尔量即可，同时判断是否存在上述情况。

code：

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

const int KIND = 10;
const int MAXN = 100005;
struct trie
{
    bool isString;      // 标志到当前节点是否是字符串
    trie* next[KIND];
    trie()
    {
        isString = false;
        for (int i = 0; i < KIND; ++i) next[i] = NULL;
    }
};

trie node[MAXN];
trie* root;
bool flag;
int k;

void Insert(string ss)
{
    trie* temp = root;
    int len = ss.size();
    for (int i = 0; i < len; ++i)
    {
        int curr = ss[i] - '0';
        if (temp->next[curr] != NULL)   // 判断前缀在前面出现的情况
        {
            if (temp->next[curr]->isString)
            {
                flag = false;
                return;
            }
        }
        else temp->next[curr] = &node[k++];
        temp = temp->next[curr];
    }
    temp->isString = true;
    for (int i = 0; i < KIND; ++i)  // 判断前缀在后面出现的情况
    {
        if (temp->next[i])
        {
            flag = false;
            return;
        }
    }
}

int main()
{
    int nCase;
    cin >> nCase;
    while (nCase--)
    {
        flag = true;
        k = 1;
        memset(node, 0, sizeof(node));
        root = &node[0];

        int n;
        cin >> n;

        string str;
        for (int i = 0; i < n; ++i)
        {
            cin >> str;
            if (flag) Insert(str);
        }
        if (flag) cout << "YES" << endl;
        else cout << "NO" << endl;
    }
    return 0;
}