HDU1251 统计难题（字典树）

最新推荐文章于 2021-11-09 00:20:03 发布

原创最新推荐文章于 2021-11-09 00:20:03 发布 · 257 阅读

0 ·

CC 4.0 BY-SA版权

数据结构---Trie 专栏收录该内容

2 篇文章

订阅专栏

本文介绍如何使用字典树高效地实现字符串前缀的插入与查询操作。通过构建字典树，可以快速确定一个给定前缀是否存在于已插入的字符串中，并统计其出现次数。文章详细解释了字典树的结构、初始化、插入和查询方法。

题意分析

用字典树，保存前缀个数，每次插入的时候加一。查询时直接查询到待查询前缀的最后一个字母，如果为前缀，那么一定保存过，否则肯定没保存过，答案就为0.

每次查询是可以认为每次插入和查询都是O(len(str))的，因为每次查询只需要沿着字典树走下来就好了。需要注意的是内存开销。
题目没说总共有多少个待查询子串，一开始一直超时，我以为是写挂了。
假设要插入 $n$ 个串，串最长为 $lenmax$ ，最差情况，需要开的空间为， $n * lenmax$ ，即认为每个字符都不同。

代码总览

#include <cstdio>
#include <cstring>
#define mem(x,y) memset(x,y,sizeof x)
#define rep(i,a,b) for (int i = a; i<=b; ++i)
using namespace std;
const int nmax = 1e6 + 10;
const int INF = 0x3f3f3f3f;
typedef long long ll;
typedef double db;
#define sigma_size 26
#define maxlength 500000
typedef struct {
    int ch[maxlength][sigma_size];
    int val[maxlength];
    int sz;
    init() { sz = 1; memset(ch[0], 0, sizeof(ch[0])); }
    inline int idx(char c) { return c - 'a'; }

    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] == 0) {
                memset(ch[sz], 0, sizeof(ch[sz]));
                ch[u][c] = sz++;
            }
            u = ch[u][c];
            val[u] ++;
        }
    }
    int search(char *s, int v) {
        int u = 0, n = strlen(s), c;
        for (int i = 0; i < n; i++) {
            c = idx(s[i]);
            if (ch[u][c] == 0) return 0;
            u = ch[u][c];
        }
        return val[u];
    }
} Trie;
Trie trie;
int main() {
    // freopen("in.txt", "r", stdin);
    char str[105];
    trie.init();
    while (gets(str) && strcmp(str, "") != 0) trie.insert(str, 1);
    while (scanf("%s", str) != EOF) printf("%d\n", trie.search(str, 1));
    return 0;
}