Hdu 2846 Repository

最新推荐文章于 2022-03-29 22:20:29 发布

转载最新推荐文章于 2022-03-29 22:20:29 发布 · 62 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/g0feng/archive/2012/09/12/2682466.html

本文介绍了一种利用字典树实现商品信息子串快速查询的方法，通过将商品信息分解并存储到字典树中，可以高效地完成对任意查询字符串的匹配，并统计匹配次数。

大意：给定你一些商品的信息save（字符串），然后输入需要查询的字符串str，如果存在一个字符串save使得str是save的子串。

思路：字典树的变形，如存在"abcd"，则"bcd", "cd", "d",都需要插入字典树中。如果存在"abab"的话，有可能"ab"会重复计数，所以需要一个特定的序列号来区分是否是同一序列中的子串。

我们可以定义一个value，若value相同，则跳过，若不同，则+1后更新value。

例如：假设总共有4个商品信息

5 5 5 5

a b a b

5 5 5

b a b

5 5

a b

4 4

a b

3 3

a b

2 2

a c

由上图可知，若序列号相同的话，则跳过。若不同，则+1，而且value需要更新避免后面的重复。

CODE：

#include <iostream>
#include <cstdio>

using namespace std;

struct Trie
{
    Trie *next[26];
    int count;
    int value;
    Trie()
    {
        for(int i = 0; i < 26; i++)
            next[i] = 0;
        count = 0;
        value = 0;
    }
};

void insert(Trie *&root, char *s, int num)
{
    int i = 0, branch = 0;
    Trie *p = root;
    if(!p)
    {
        p = new Trie();
        root = p;
    }
    while(s[i])
    {
        branch = s[i]-'a';
        if(!p->next[branch]) p->next[branch] = new Trie();
        p = p->next[branch];
        if(p->value != num)
        {
            p->count++;
            p->value = num;
        }
        i++;
    }
}

int find(Trie *root, char *s)
{
    int i = 0, branch = 0, ans;
    Trie *p = root;
    if(!p) return 0;
    while(s[i])
    {
        branch = s[i]-'a';
        if(!p->next[branch]) return 0;
        p = p->next[branch];
        ans = p->count;
        i++;
    }
    return ans;
}

int main()
{
    int Q;
    scanf("%d", &Q);
    char str[21];
    Trie *root = NULL;
    while(Q--)
    {
        scanf("%s", str);
        for(int i = 0; i < strlen(str); i++)
            insert(root, str+i, Q+1);                //序列号
    }
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%s", str);
        printf("%d\n", find(root, str));
    }
    return 0;
}

转载于:https://www.cnblogs.com/g0feng/archive/2012/09/12/2682466.html