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";
    }
}
相关推荐
金智维科技官方5 小时前
常见的大模型分类
人工智能·算法·ai·语言模型·数据挖掘
yzzzzzzzzzzzzzzzzz5 小时前
leetcode热题——有效的括号
算法·
崎岖Qiu5 小时前
leetcode1343:大小为K的子数组(定长滑动窗口)
java·算法·leetcode·力扣·滑动窗口
Shun_Tianyou6 小时前
Python Day25 进程与网络编程
开发语言·网络·数据结构·python·算法
Giser探索家7 小时前
什么是2米分辨率卫星影像数据?
大数据·人工智能·数码相机·算法·分类·云计算
jz_ddk8 小时前
[科普] AI加速器架构全景图:从GPU到光计算的算力革命
人工智能·学习·算法·架构
曦月逸霜8 小时前
内部排序算法总结(考研向)
考研·算法·排序算法
仪器科学与传感技术博士9 小时前
Matplotlib库:Python数据可视化的基石,发现它的美
开发语言·人工智能·python·算法·信息可视化·matplotlib·图表可视化
草莓熊Lotso10 小时前
【洛谷题单】--分支结构(二)
c语言·c++·经验分享·其他·刷题
snowfoootball10 小时前
2025 蓝桥杯C/C++国B 部分题解
c语言·c++·笔记·学习·贪心算法·蓝桥杯