HDU 1671 Phone List（Trie）

题目链接

题意

      给定n个电话号码，看看是否存在某些号码是别的号码的前缀，若存在，输出“NO"，否则输出”YES“。

数据范围的一些限定

1 <= n <= 10000

电话号码长度不超过10个数字

思路

      暴力的做法先输入完所有的号码，然后从第一个开始到最后一个，一个一个检测当前号码是否是其他号码的前缀，判断当前号码是否是别的号码的前缀的方法大家应该都懂，就是逐个数字对比直到对比完当前号码为止。

      看到这种一个字符串跟一堆字符串逐个进行比较的操作的时候，就会想起Trie。要使用Trie的话，必须解决一个问题，怎么判断一个字符串是Trie中的某个字符串的前缀。如果字符串tmp是Trie中某个字符串的前缀，那么tmp对应的单词结点必然会有子结点，可以依此来判断。

      再看一看题目，题目要求只要有一个号码是别的号码的前缀，就输出”NO“。那么就不用把所有输入的电话号码都插入Trie了，可以边输入边判断，把当前输入的号码插入Trie中并检查Trie中是否存在一个号码是其他Trie中任一号码的前缀，存在则以后的输入不再插入Trie中，要输出什么就很容易判断了。

      边输入边判断的方法有一个要注意的地方。向Trie插入号码的时候要注意检查两件事。向Trie插入号码b时，要检查路径上已存在的结点中是否有单词结点，若有，则Trie中必然存在一个号码是号码b的前缀，反之则当前Trie中任何的非b号码都不是b的前缀。插入b的单词结点时还要检查该单词结点是否已被创建过（创建过是指有单词路过这个点，即child数组相应的值不为0），若该结点已被创建过，则b必然是Trie中某个 （至少一个，有多少个不能从子结点的个数来判断，发散一下，与题目无关） 号码的前缀，反之Trie中任何的非b号码都不以b为前缀。

      更具体一些的细节请自己看下面附上的代码吧。

      以下是代码，各种多余头文件宏定义什么的请无视，在vimrc里写好了，懒得删掉了，可以视为我在装13。如果在其中发现有错误，请指出，发表此文部分原因是希望有人能够指出我可能存在的错误。如果有人有更好的解法，请不吝赐教。

/*
 * Author:  Fiend
 * Created Time:  2013/4/8 15:36:16
 * File Name: test.cpp
 */
#include <iostream>
#include <cstdio>
#include <cstddef>
#include <cmath>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <bitset>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <cctype>

#define ST size_type
#define PB push_back
#define LL long long
#define SIGMA_SIZE 15
#define NODE_SIZE 100005

using std::cin;
using std::cout;
using std::endl;
using std::string;
using std::bitset;
using std::vector;
using std::pair;
using std::swap;
using std::sort;
using std::max;
using std::min;

const int inf = 0x3fffffff;

typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<int>::iterator vit;

class Trie {
    private:
        int child[NODE_SIZE][SIGMA_SIZE];
        bool val[NODE_SIZE];	//记录该结点是否为单词结点
        int cnt;    //结点总数
    public:
        //初始时只有一个根结点
        Trie () {
            cnt = 1;
            memset (child[0], 0, sizeof (child[0]));
        }

        //将trie重置为只有一个根结点
        void clear () {
            cnt = 1;
            memset (child[0], 0, sizeof (child[0]));
        }

        //返回字符ch的编号
		//根据字符集的不同进行调整
        int getIndex (char ch) {
            return ch -'0';
        }

        /*
         * 插入字符串src
         * src是string类型，插入前不用查询是否存在
         * 对象做实参要加引用，不然每次一调用函数就要复制一个对象，很影响效率
		 * 若src在Trie中存在其前缀或者Trie中存在字符串以src为前缀，则返回true，否则返回false
         */
        bool insert (string &src) {
            int parent = 0, index;
			bool rtn = false;
            for (string::size_type i = 0; i != src.size(); ++i) {
                index = getIndex (src[i]);
                if (child[parent][index] == 0) {    //结点不存在
                    child[parent][index] = cnt;     //新建结点
                    memset (child[cnt], 0, sizeof(child[cnt]));
                    val[cnt] = false;	//非单词结点
                    ++cnt;
                } else {//该结点已被创建过
					//插入号码时的两项检查
					if (i + 1 == src.size() || val[child[parent][index]])
						return true;
				}
                parent = child[parent][index];  //往下走
            }
            val[parent] = true;		//单词结点
			return rtn;
        }
};

Trie t;

int main () {
	int T, n;
	string tmp;
	bool ans;
	cin >> T;
	while (T--) {
		t.clear ();
		ans = false;
		cin >> n;
		for (int i = 1; i <= n; ++i) {
			cin >> tmp;
			if (ans == false)
				ans = t.insert(tmp);
		}
		if (ans)
			cout << "NO" << endl;
		else
			cout << "YES" << endl;
	}
	return 0;
}