hdu 1671 phone list（字典树）

题意：

判断输入的电话号码的前缀，是否是其他电话号码，如果是其他号码就输出NO，否则输出YES。

解析：

这题的思路很简单，直接利用Trie树求解。
先把所有的电话号码从1到n进行标号，并插入Trie树中，然后利用Trie树查找每个电话号码，如果在查找的过程中遇到val != 0就return，判断返回的结果是不是当前的电话号码的编号，如果不是就return false。

$my$ $code$

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int sigma_size = 10;
const int maxnode = 100005;
const int MAXN = 10005;

int ch[maxnode][sigma_size];
int val[maxnode];
int sz;

struct Trie {
    Trie() { sz = 1; memset(ch[0], 0, sizeof(ch[0])); }
    int idx(char c) { return c - '0'; }

    void insert(char *s, int v) {
        int u = 0, n = strlen(s);
        for(int i = 0; i < n; i++) {
            int c = idx(s[i]);
            if(!ch[u][c]) {
                memset(ch[sz], 0, sizeof(ch[sz]));
                val[sz] = 0;
                ch[u][c] = sz++;
            }
            u = ch[u][c];
        }
        val[u] = v;
    }

    int query(char *s) {
        int u = 0, n = strlen(s);
        for(int i = 0; i < n; i++) {
            int c = idx(s[i]);  
            u = ch[u][c];
            if(val[u]) return val[u];
        }
    }
};

int n;
char str[MAXN][15];

bool judge(Trie trie) {
    for(int i = 1; i <= n; i++) {
        int ret = trie.query(str[i]);
        if(ret != i && ret != 0)
            return false;
    }
    return true;
}

int main() {
    int T;
    scanf("%d", &T);
    while(T--) {
        Trie trie;
        scanf("%d", &n);
        for(int i = 1; i <= n; i++) {
            scanf("%s", str[i]);
            trie.insert(str[i], i);
        }
        printf("%s\n", judge(trie) ? "YES" : "NO");
    }
    return 0;
}