연결 리스트로 구현된 리스트 ADT 테스트 프로그램

#include <stdio.h>
#include <stdlib.h>
#include <malloc.h>
#include <limits.h>
 
#define FALSE 0
#define TRUE 1
 
typedef int element;
typedef struct ListNode {
    element data;
    struct ListNode *link;
} ListNode;
 
typedef struct {
    ListNode *head;        // 헤드 포인터
    int length;                // 노드의 개수
} LinkedListType;
 
void error(char *message) {
    fprintf(stderr, "%s\n", message);
    exit(1);
}
 
void init(LinkedListType *list) {
    if (list == NULL) return;
    list->length = 0;
    list->head = NULL;
}
 
void insert_node(ListNode **phead, ListNode *p, ListNode *new_node) {
    if (*phead == NULL) {
        new_node->link = NULL;
        *phead = new_node;
    }
    else if (p == NULL) {
        new_node->link = *phead;
        *phead = new_node;
    }
    else {
        new_node->link = p->link;
        p->link = new_node;
    }
}
 
void remove_node(ListNode **phead, ListNode *p, ListNode *removed) {
    if (p == NULL) {
        *phead = (*phead)->link;
    }
    else {
        p->link = removed->link;
    }
    free(removed);
}
 
ListNode *get_node_at(LinkedListType *list, int pos) {
    int i;
    ListNode *tmp_node = list->head;
    if (pos < 0) return NULL;
    for (i = 0; i < pos; i++) {
        tmp_node = tmp_node->link;
    }
    return tmp_node;
}
 
void add(LinkedListType *list, int position, element data) {
    ListNode *p;
    if ((position >= 0) && (position <= list->length)) {
        ListNode *node = (ListNode *)malloc(sizeof(ListNode));
        if (node == NULL) error("메모리 할당에러");
        node->data = data;
        p = get_node_at(list, position - 1);
        insert_node(&(list->head), p, node);
        list->length++;
    }
}
 
void add_last(LinkedListType *list, element data) {
    add(list, get_length(list), data);
}
 
void add_first(LinkedListType *list, element data) {
    add(list, 0, data);
}
 
int is_empty(LinkedListType *list) {
    if (list->head == NULL) return 1;
    else return 0;
}
 
int get_length(LinkedListType *list) {
    return list->length;
}
 
void delete(LinkedListType *list, int pos) {
    if (!is_empty(list) && (pos >= 0) && (pos < list->length)) {
        ListNode *p = get_node_at(list, pos - 1);
        ListNode *removed = get_node_at(list, pos);
        remove_node(&(list->head), p, removed);
        list->length--;
    }
}
 
element get_entry(LinkedListType *list, int pos) {
    ListNode *p;
    if (pos >= list->length) error("위치 오류");
    p = get_node_at(list, pos);
    return p->data;
}
 
void clear(LinkedListType *list) {
    int i;
    for (i = list->length - 1; i >= 0; i--) {
        delete(list, i);
    }
}
 
void display(LinkedListType *list) {
    int i;
    ListNode *node = list->head;
    printf("( ");
    for (i = 0; i < list->length; i++) {
        printf("%d ", node->data);
        node = node->link;
    }
    printf(" )\n");
}
 
int is_in_list(LinkedListType *list, element item) {
    ListNode *p;
    p = list->head;        // 헤드 포인터에서부터 시작한다.
    while ((p != NULL)) {
        // 노드의 데이터가 item이면
        if (p->data == item) {
            break;
        }
        p = p->link;
    }
    if (p == NULL) return FALSE;
    else return TRUE;
}
 
int main() {
    LinkedListType list1;
 
    init(&list1);
    add(&list1, 0, 20);
    add_last(&list1, 30);
    add_first(&list1, 10);
    add_last(&list1, 40);
 
    // list1 = (10, 20, 30, 40)
    display(&list1);
 
    // list1 = (10, 20, 30)
    delete(&list1, 3);
    display(&list1);
 
    // list1 = (20, 30)
    delete(&list1, 0);
    display(&list1);
 
    printf("%s\n", is_in_list(&list1, 20) == TRUE ? "성공" : "실패");
    printf("%d\n", get_entry(&list1, 0));
}

코드 출처 : C언어로 쉽게 풀어쓴 자료구조 (천인국 외 지음, 생능출판사)