码迷,mamicode.com
首页 > Windows程序 > 详细

acwing 142. 前缀统计

时间:2019-10-12 10:42:00      阅读:78      评论:0      收藏:0      [点我收藏+]

标签:题解   har   iostream   break   res   msu   std   答案   字符   

题面:

给定N个字符串S1,S2SNS1,S2…SN,接下来进行M次询问,每次询问给定一个字符串T,求S1S1~SNSN中有多少个字符串是T的前缀。

输入字符串的总长度不超过106106,仅包含小写字母。

输入格式

第一行输入两个整数N,M。

接下来N行每行输入一个字符串SiSi。

接下来M行每行一个字符串T用以询问。

输出格式

对于每个询问,输出一个整数表示答案。

每个答案占一行。

输入样例:

3 2
ab
bc
abc
abc
efg

输出样例:

2
0
题解:
这是比较裸的trie题
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int son[1000010][26],cnt[1000010],idx;
char str[1000010];
void insert()
{
    int p=0;
    for(int i=0;str[i];i++)
    {
        int &s=son[p][str[i]-a];
        if(!s)s=++idx;
        p=s;
    }
    cnt[p]++;
}
int query()
{
    int p=0;int res=0;
    for(int i=0;str[i];i++)
    {
        int &s=son[p][str[i]-a];
        if(!s)break;
        p=s;
        res+=cnt[p];
    }
    return res;
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    while(n--)
    {
        scanf("%s",str);
        insert();
    }
    while(m--)
    {
        scanf("%s",str);
        printf("%d\n",query());
    }
    return 0;
}

 

acwing 142. 前缀统计

标签:题解   har   iostream   break   res   msu   std   答案   字符   

原文地址:https://www.cnblogs.com/flyljz/p/11659965.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!