정적 할당을 이용한 Trie

아래와 같이 triePool을 만들어두고 동적 할당이 아닌 정적 할당을 해주고 

alloc 이라는 함수를 만들어 해당하는 배열의 인덱스의 주소를 넘기면 우리가 흔히 하는 new를 이용 할 수 있다.

그리고 delete 대신 trieIdx = 0을 통해 새로이 트라이 배열을 사용 할 수 있어 delete 효과를 볼 수 있다.

소스 코드 : 

#include <iostream>
 
#define MAX_SIZE 26
 
struct Trie {
  Trie *next[MAX_SIZE];
  bool isEnd;
  int cnt;
};
 
Trie triePool[1000002];
int trieIdx;
 
Trie *root;
 
void init() {
  root = nullptr;
  trieIdx = 0;
}
 
Trie *alloc() {
  for (int i = 0; i < MAX_SIZE; i++)
    triePool[trieIdx].next[i] = nullptr;
  triePool[trieIdx].isEnd = triePool[trieIdx].cnt = 0;
 
  return &triePool[trieIdx++];
}
void insert(int buffer_size, char *buf) {
  if(root == nullptr)
    root = alloc();
 
  Trie *cur = root;
 
  int len = buffer_size;
 
  for (int i = 0; i < len; i++) {
    int ch = buf[i] - 'a';
    if (cur->next[ch] == nullptr)
      cur->next[ch] = alloc();
    
    cur = cur->next[ch];
    cur->cnt++;
  }
  cur->isEnd = true;
}
 
int query(int buffer_size, char *buf) {
  if (root == nullptr)
    return 0;
 
  Trie *cur = root;
  int len = buffer_size;
 
  for (int i = 0; i < len; i++) {
    int ch = buf[i] - 'a';
    if (cur->next[ch] == nullptr)
      return 0;
    cur = cur->next[ch];
  }
 
  return cur->cnt;
}