hdu 1671 Phone List 字典树

最新推荐文章于 2019-09-12 17:13:21 发布

原创最新推荐文章于 2019-09-12 17:13:21 发布 · 939 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#list #null #struct #电话

数据结构专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种使用字典树（Trie）的数据结构来解决电话号码是否为其他号码前缀的问题。通过创建字典树并遍历输入的电话号码，可以有效地检查是否存在一个号码作为另一个号码的前缀。文章提供了完整的C++实现代码，并强调了释放内存的重要性以避免内存泄漏。

题意：判断是否有一电话号码是另一个的前缀。

思路：建立字典树，参考维基百科：http://zh.wikipedia.org/wiki/Trie

需要注意的是最后释放内存，否则会MLE。

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

const int N = 10;

struct Trie {
    bool flag;
    Trie *next[N];
    Trie() {
        flag = true;
        int i;
        for (i = 0; i < N; ++i)
            next[i] = NULL;
    }
};

bool flag;

void creat(Trie *root, char s[]) {
    int i, len, b;
    len = strlen(s);
    Trie *p = root;
    for (i = 0; i < len; ++i) {
        b = s[i] - '0';
        if (p->next[b]) {
            if (p->next[b]->flag == false || i == len-1) {
                flag = true;
                return;
            }
        }
        else
            p->next[b] = new Trie();
        p = p->next[b];
    }
    p->flag = false;
}

void freedom(Trie *&root) {
    int i;
    for (i = 0; i < N; ++i)
        if (root->next[i] != NULL)
            freedom(root->next[i]);
    free(root);
}

int main()
{
    int t, n;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        flag = false;
        char s[19];
        Trie *root = NULL;
        root = new Trie();
        while (n--) {
            scanf("%s", s);
            if (flag)
                continue;
            creat(root, s);
        }
        if (flag)
            printf("NO\n");
        else
            printf("YES\n");
        freedom(root);
    }
    return 0;
}