【发布时间】:2018-06-05 05:35:40
【问题描述】:
我正在尝试实现 trie 数据结构并编写以下代码
#include<iostream>
#include<string>
#include<map>
#include<vector>
using namespace std;
struct trie_node{
map<char, trie_node> child;
int pos=-1;
};
map<char, trie_node> baap;
void lex_insert(string s, int pos){ // lexiographically insert into map
trie_node t = baap[s[0]];
for(int i=1; i<s.size(); i++){
t = t.child[s[i]];
}
t.pos = pos;
}
int find(string s, int r){ // find in trie structure
trie_node t = baap[s[0]];
int pos = t.pos;
for(int i=1; i<s.size(); i++){
if(t.child.find(s[i])!=t.child.end()){
t = t.child[s[i]];
if(t.pos<=r)
pos = t.pos;
}
else
return pos;
}
return pos;
}
void ans(int &found, string s){ // find lexiographically matching prefix
trie_node t = baap[s[0]];
if(found<0){
while(t.pos<0){
auto x = t.child.begin();
t = x->second;
}
found = t.pos;
}
}
int main(){
int n; cin>>n;
vector<string> vs(n);
for(int i=0; i<n; i++){
cin>>vs[i];
lex_insert(vs[i],i);
}
int found = 0;
int q; cin>>q;
for(int i=0; i<q; i++){
int r; string p;
cin>>r>>p;
found = find(p,r);
ans(found, p);
cout<<vs[found]<<'\n';
}
}
请关注lex_insert()
代码执行没有错误,但当我尝试取消引用地图时 - baap,出现分段错误。我想这是因为我没有在lex_insert 的每个级别上构建新节点,正如我在一些代码中看到的那样。但是,地图应该进行动态分配。有人可以解释一下,为什么我无法访问地图的元素 - baap ?
注意 - 它不是一个精确的 trie 实现,而是派生自它
【问题讨论】:
标签: c++ dictionary trie