首页 > 编程语言 > 详细

题解|《算法竞赛进阶指南》 前缀统计

时间:2019-08-30 11:00:01      阅读:127      评论:0      收藏:0      [点我收藏+]

题目描述
给定N个字符串S_{1},S_{2}...S_{n}S1?,S2?...Sn? ,接下来进行M次询问,每次询问给定一个字符串T,求S1~Sn 中有多少个字符串是T的前缀。输入字符串的总长度不超过10^6 ,仅包含小写字母。

输入描述:
第一行两个整数N,M。接下来N行每行一个字符串Si。接下来M行每行一个字符串表示询问。

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

思路
一看到字符串的前缀,我们就应该想到字典树,和字典一样的前缀树.这道题是字典树很经典的一道题也没什么好说的。

完整C++版AC代码

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1000010;

int n,m;
int son[N][26], cnt[N], idx;
char str[N];

void insert() {
    int p = 0;
    for (int i = 0; str[i]; i++) {
        int s = str[i] - a;
        if (!son[p][s]) son[p][s] = ++idx;
        p = son[p][s];
    }
    cnt[p]++;
}

int search() {
    int p = 0, ans = 0;
    for (int i = 0; str[i]; i++) {
        int s = str[i] - a;
        if (!son[p][s]) break;
        p = son[p][s];
        ans += cnt[p];
    }
    return ans;
}

int main() {
    ios::sync_with_stdio;

    cin >> n >> m;
    while (n--) {
        cin >> str;
        insert();
    }
    while (m--) {
        cin >> str;
        int ans = search();
        cout << ans << endl;
    }
    return 0;
}

 

题解|《算法竞赛进阶指南》 前缀统计

原文:https://www.cnblogs.com/Mashiro-zBlog/p/11433436.html

(1)
(1)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!