Hdu 2846 Repository

 

大意：给定你一些商品的信息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

         5

         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;
}