Codeforces 4C Runtime Ошибка

Question

Привет, я использую реализацию Trie для решения этой проблемы, но на серверах codeforces я получаю ошибку времени выполнения, и на моем компьютере, и на Ideone это нормально. Есть идеи? http://codeforces.com/contest/4/problem/C https://ideone.com/dvRL3b

#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>


using namespace std;
#define MAXN  ('z'-'a'+2)
#define VALUE(c) ( (c)-'a' )

struct Trie
{
    Trie *m[MAXN];
    int count;
}trie;

int insert(Trie *  t, char * w){
    if(w[0]=='\0'){
        t->count++;
        return t->count;
    }
    if(t->m[VALUE(w[0])] == NULL){
        t->m[VALUE(w[0])] =(Trie *) malloc(sizeof(Trie));
    }
    return insert(t->m[VALUE(w[0])] , w+1);
}   


int main(){

    int t;
    scanf("%d",&t);

    for (int i = 0; i < t; ++i)
    {
        char *w = NULL;
        w = (char *)malloc(sizeof(char));
        scanf("%s",w);
        //printf("%s\n", w);
        int resp = insert(&trie,w); 
        if(resp > 1){
            printf("%s%d\n", w,resp-1);
        }else{
            printf("OK\n");
        }
    }


    return 0;

}

Answer 1

Эта строка:

w = (char *)malloc(sizeof(char));

выделит только один байт памяти.

Запросы в этой проблеме не более 32 символов, поэтому попробуйте:

w = (char *)malloc(sizeof(char)*33); // include 1 extra byte for zero termination byte

Кроме того, вы, вероятно, должны использовать calloc при выделении памяти для каждого нового Trie, т.е.

t->m[VALUE(w[0])] =(Trie *) malloc(sizeof(Trie));

в

t->m[VALUE(w[0])] =(Trie *) calloc(sizeof(Trie),1);

в противном случае указатели не будут очищены.