迷之好奇

y131405y

于 2021-03-13 19:47:08 发布

阅读量72

点赞数

分类专栏：字典树文章标签：数据结构算法

本文链接：https://blog.csdn.net/a123321a1233211/article/details/114759805

版权

字典树专栏收录该内容

2 篇文章 0 订阅

订阅专栏

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

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

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

Input
多组输入。
对于每组数据

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

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

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

接下来的m行。

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

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

输出示例

1
0
1

答案代码

#include <bits/stdc++.h>
#include <stdio.h>
#include <string.h>
using namespace std;
typedef struct node
{

    int id;
    struct node *next[10];
    int f;
}NODE;
NODE T[1000000];
int top;

NODE *creat()
{

    NODE *p;
    int i;
    p=&T[top++];
    for(i=0;i<=9;i++)
    {
        p->next[i]=NULL;

    }
    p->id=0;
    p->f=0;
    return p;
}
void insertt(NODE *root,char s[])
{

    NODE *p=root;
    int i;
    for(i=0;s[i]!='\0';i++)
    {

        int x=s[i]-'0';
        if(p->next[x]!=NULL)
        {

            p=p->next[x];
        }
        else
        {

            p->next[x]=creat();
            p=p->next[x];
        }
        p->id++;
    }
    p->f++;
}

int check(NODE *t,char s[])
{

    NODE *p=t;
    int i;
    for(i=0;s[i]!='\0';i++)
    {

        int x=s[i]-'0';
        if(p->next[x]!=NULL)
        {

            p=p->next[x];
        }
        else
        {

            return 0;
        }
    }
    int k=p->id;
    if(p->f!=0)
        k-=p->f;
    return k;
}
int main()
{

    int i,j,n,m;
    char s[100010];
    char s1[100010];
    while(~scanf("%d",&n))
    {

        NODE *t;
        top=0;
        t=(NODE*)malloc(sizeof(NODE));
        t=creat();
        for(i=0;i<n;i++)
        {

            scanf("%s",s1);
            int len=strlen(s1);
            for(j=0;j<len;j++)
            {

                s[j]=s1[len-1-j];
            }
            s[len]='\0';
            insertt(t,s);
        }
        scanf("%d",&m);
        for(i=0;i<m;i++)
        {

            scanf("%s",s1);
            int len=strlen(s1);
            for(j=0;j<len;j++)
            {

                s[j]=s1[len-1-j];
            }
            s[len]='\0';
            int k=check(t,s);
            printf("%d\n",k);

        }
    }
    return 0;
}