基礎的trie樹題

#include<bits/stdc++.h> using namespace std; const int N=1e6+10; int son[N][26],cnt[N],idx; int n,m; void insert(string s) {int p=0;for(int i=0;i<s.size();i++){int u=s[i]-'a';if(!son[p][u]) son[p][u]=++idx;p=son[p][u];}cnt[p]++; } int query(string s) {int p=0,sum=0;for(int i=0;i<s.size();i++){int u=s[i]-'a';if(!son[p][u]) return sum;p=son[p][u];sum+=cnt[p];}return sum; } int main(void) {cin>>n>>m;for(int i=0;i<n;i++){string s; cin>>s;insert(s);}for(int i=0;i<m;i++){string s; cin>>s;cout<<query(s)<<endl;}return 0; }

總結

以上是生活随笔為你收集整理的142. 前缀统计【trie】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。