使用地图实现trie而不创建新节点

时间:2018-06-05 05:35:41

标签: c++ dictionary trie

我试图实现trie数据结构&写了foll代码

#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实现,而是源于它

1 个答案:

答案 0 :(得分:1)

问题的一个可能原因:

trie_node t = baap[s[0]];

此处t将是baap[s[0]]副本。您对t所做的任何更改都不会反映在原始内容中。

由于您稍后在循环中重新分配到t(再次制作副本),您无法使用引用(引用无法反弹)。相反,你必须使用指针。要么在地图中存储指针,要么使用address-of运算符来获取指针:

trie_node* t = &baap[s[0]];  // Get a pointer to the node

至于崩溃,如果上述方法没有解决,那么你应该learn how to debug your programs。特别是对于崩溃,您应该学习如何使用调试器来捕获崩溃的操作以及如何找到代码中的位置。