SDUTOJ 3039 迷之好奇 静态字典树

迷之好奇

Time Limit: 2000MS Memory limit: 65536K

题目描述

FF得到了一个有n个数字的集合。不要问我为什么，有钱，任性。

FF很好奇的想知道，对于数字x，集合中有多少个数字可以在x前面添加任意数字得到。

如，x = 123，则在x前面添加数字可以得到4123，5123等。

输入

 多组输入。

对于每组数据

首先输入n(1<= n <= 100000)。

接下来n行。每行一个数字y(1 <= y <= 100000)代表集合中的元素。

接下来一行输入m(1 <= m <= 100000),代表有m次询问。

接下来的m行。

每行一个正整数x(1 <= x <= 100000)。

输出

 对于每组数据，输出一个数字代表答案。

示例输入

3
12345
66666
12356
3
45
12345
356

示例输出

1
0
1

提示

 

来源

 zmx

需要用静态字典树来写，动态开内存的会MLE。。。。。

#include <stdio.h>
#include <string.h>

struct node
{
    int flag;
    int num;
    int next[10];
}ls[600010];

int top;

void creat(char *s,int xb)
{
    int len = strlen(s);
    for(int i = len - 1;i >= 0;i--)
    {
        int zh = s[i] - '0';
        if(!ls[xb].next[zh])
        {
            ls[top].flag = 0;
            ls[top].num = 1;
            for(int j = 0;j < 10;j++)
                ls[top].next[j] = 0;
            ls[xb].next[zh] = top;
            xb = top++;
        }
        else
        {
            xb = ls[xb].next[zh];
            ls[xb].num++;
        }
    }
    ls[xb].flag++;
}

int nfind(char *s,int xb)
{
    int mn = 0x3f3f3f3f;
    int len = strlen(s);
    for(int i = len - 1;i >= 0;i--)
    {
        int zh = s[i] - '0';
        if(ls[xb].next[zh])
        {
            xb = ls[xb].next[zh];
            if(mn > ls[xb].num)
                mn = ls[xb].num;
        }
        else
            return -1;
    }
    return mn - ls[xb].flag;
}

int main()
{
    int n,m;
    char s1[10];
    while(~scanf("%d",&n))
    {
        top = 0;
        for(int i = 0;i < 10;i++)
            ls[top].next[i] = 0;
        top++;
        for(int i = 0;i < n;i++)
        {
            scanf("%s",s1);
            creat(s1,0);
        }
        scanf("%d",&m);
        for(int i = 0;i < m;i++)
        {
            scanf("%s",s1);
            int fh = nfind(s1,0);
            if(fh == -1)
                printf("0\n");
            else
                printf("%d\n",fh);
        }
    }
    return 0;
}