7. 5052 전화번호 목록

원본 출처 : https://www.acmicpc.net/problem/5052

5052번: 전화번호 목록

문제 풀이 : 

#include <stdio.h>
#include <malloc.h>
#define NUM(X) ((X)-'0')
typedef struct _node {
    int last;
    struct _node *next[10];
}NODE;
NODE root;
int N;
void insert_node(NODE *p, char *str) {
    //int i;
    //NODE *cur = p;
 
    //for (i = 0; i < str[i]; i++) {
    //  NODE *newNode = calloc(1, sizeof(NODE));
    //  int t;
    //  t = (int)str[i] - 48;
    //  if (cur == &root) {
    //      //char 1 = int 49
    //      newNode->last = 1;
    //      cur->next[t] = newNode;
    //  }
    //  else {
    //      while (cur->next[t] != NULL) {
    //          cur->last = 0;
    //          cur = cur->next[t];
    //      }
    //      newNode->last = 1;
    //      cur->next[t] = newNode;
    //  }
    //}
    int i, n;
    NODE *new;
    for (i = 0; str[i]; i++) {
        n = NUM(str[i]);    //정수화
        if (p->next[n] == 0) {
            new = calloc(1, sizeof(NODE));
            p->next[n] = new;
        }
        p = p->next[n];
    }
    p->last = 1;
}
int search_node(NODE *p, char *str) {
    int i, n;
    NODE *new;
    for (i = 0; str[i]; i++) {
        n = NUM(str[i]);                //정수화
        if (p->last) return 1;           //기존문자열이 먼저 끝나면 현문자열이 일부분임
        if (p->next[n] == 0) return 0;   //내문자가 없다면 일부분 아님
        p = p->next[n];
    }
    if (p->last == 0) return 1;
    else return 0;
}
void delete_node(NODE *p) {
    int i;
    for (i = 0; i < 10; i++) {
        if (p->next[i]) {
            delete_node(p->next[i]);
            free(p->next[i]);
            p->next[i] = 0;
        }
    }
}
int main() {
    int ti, T;
    scanf("%d", &T);
    while (T--) {
        int i, flag = 0;
        char str[21];
        scanf("%d", &N);
        for (i = 0; i < N; i++) {
            scanf("%s", str);
            if (i != 0)flag |= search_node(&root, str);
            insert_node(&root, str);
        }
        if (flag) printf("NO\n");
        else printf("YES\n");
        delete_node(&root);
    }
    return 0;
}