P3879 [TJOI2010] 阅读理解- 字典树

题面

分析

将所有单词存入字典树,重点值怎么判断在哪一行出现过,对于字典树查询的判断字符串是否存在的数组可以开成二维,也就是在查询到某个字符串存在后,再通过循环判断每一层是否存在。

代码
cpp 复制代码
#include <bits/stdc++.h>

#define int long long

using namespace std;

const int N = 5e5 + 10;

int son[N][30];
bitset<1010> vis[N];
int idx;
int n;

void insert(string s, int i) {
    int p = 0;
    for(int i = 0; i < s.size(); i ++) {
        int c = s[i] - 'a';
        if(!son[p][c]) son[p][c] = ++ idx;
        p = son[p][c];
    }
    vis[p][i] = 1;
}

vector<int> query(string s) {
    int p = 0;
    vector<int> ans;
    for(int i = 0; i < s.size(); i ++) {
        int c = s[i] - 'a';
        if(!son[p][c]) return ans;
        p = son[p][c];
    }
    for(int i = 1; i <= n; i ++) {
        if(vis[p][i]) ans.push_back(i);
    }
    return ans;
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    cin >> n;
    for(int i = 1; i <= n; i ++) {
        int l;
        cin >> l;
        for(int j = 0; j < l; j ++) {
            string s;
            cin >> s;
            insert(s, i);
        }
    }
    int m;
    cin >> m;
    while(m --) {
        string s;
        cin >> s;
        vector<int> ans = query(s);
        for(int i = 0; i < ans.size(); i ++) cout << ans[i] << ' ';
        cout << "\n";
    }
}
相关推荐
ゞ 正在缓冲99%…4 分钟前
leetcode22.括号生成
java·算法·leetcode·回溯
小卡皮巴拉15 分钟前
【力扣刷题实战】矩阵区域和
开发语言·c++·算法·leetcode·前缀和·矩阵
WG_1723 分钟前
第五章.图论
算法·图论
Pacify_The_North27 分钟前
【C++进阶三】vector深度剖析(迭代器失效和深浅拷贝)
开发语言·c++·windows·visualstudio
神里流~霜灭34 分钟前
蓝桥备赛指南(12)· 省赛(构造or枚举)
c语言·数据结构·c++·算法·枚举·蓝桥·构造
扫地的小何尚39 分钟前
NVIDIA工业设施数字孪生中的机器人模拟
android·java·c++·链表·语言模型·机器人·gpu
小杨爱学习zb41 分钟前
学习总结 网格划分+瞬态求解设置
笔记·学习·算法
Zfox_41 分钟前
【C++项目】从零实现RPC框架「四」:业务层实现与项目使用
linux·开发语言·c++·rpc·项目
我想吃余44 分钟前
【C++篇】类与对象(上篇):从面向过程到面向对象的跨越
开发语言·c++
双叶8361 小时前
(C语言)单链表(1.0)(单链表教程)(数据结构,指针)
c语言·开发语言·数据结构·算法·游戏