1

我正在尝试实现TopCoder页面上显示的 trie。我正在对其进行一些修改以存储用户的电话号码。我遇到分段错误。有人可以指出错误。

 #include<iostream>
 #include<stdlib.h>
 using namespace std;

 struct node{
int words;
int prefix;
long phone;
struct node* children[26];
 };

struct node* initialize(struct node* root) {
    root = new (struct node);   
    for(int i=0;i<26;i++){
    root->children[i] = NULL;
    }
    root->word = 0;
    root->prefix = 0;
    return root;
 }

int getIndex(char l) {
    if(l>='A' && l<='Z'){
    return l-'A';
    }else if(l>='a' && l<='z'){
    return l-'a';
    }
 }

  void add(struct node* root, char * name, int data) {

    if(*(name)== '\0') {
        root->words = root->words+1;
        root->phone = data;
    } else {        
        root->prefix = root->prefix + 1;
        char ch = *name;
        int index = getIndex(ch);
        if(root->children[ch]==NULL)    {
            struct node* temp = NULL;
            root->children[ch] = initialize(temp);
        }
        add(root->children[ch],name++, data);
    }
 }

 int main(){
     struct node* root = NULL;
     root = initialize(root);
     add(root,(char *)"test",1111111111);
     add(root,(char *)"teser",2222222222);
         cout<<root->prefix<<endl;
     return 0;
  }

在进行建议的更改后添加了一个新功能:

 void getPhone(struct node* root, char* name){
     while(*(name) != '\0' || root!=NULL) {
         char ch = *name;
         int index = getIndex(ch);
         root = root->children[ch];
         ++name;
     }
     if(*(name) == '\0'){
         cout<<root->phone<<endl;
     }
 }
4

1 回答 1

1

改变这个:

add(root->children[ch], name++, data);
// ---------------------^^^^^^

对此:

add(root->children[ch], ++name, data);
// ---------------------^^^^^^

这段代码中的其余问题我留给您,但这是您启动调用堆栈的原因。

EDIT OP 要求进一步分析,虽然我通常不这样做,但这是一个相当简单的扩展应用程序。

这是在几个地方完成的:

 int index = getIndex(ch);
 root = root->children[ch];
 ... etc. continue using ch instead of index

它引出了一个问题:“为什么我们只是要求一个我们立即忽略并使用 char 的索引?” 这是在add()和中完成的getPhone()。您应该在为数组index内的所有窥视计算它之后使用它。children[]

此外,该initialize()函数需要进行修改或彻底抛弃,以支持基于构造函数的解决方案,该代码真正属于该解决方案。最后,如果这个 trie 应该跟踪生成的单词的使用计数和每个级别参与的前缀,我不清楚为什么你需要单词和前缀计数器,但在任何一种情况下都应该更新你的递归体面add()计数器在反向递归上将它们撞起来。

于 2013-04-20T19:31:28.157 回答