Programming/C C++

[DS] Insert a node at the head of a linked list

ElAsJay 2020. 8. 30. 02:42
728x90

www.hackerrank.com/challenges/insert-a-node-at-the-head-of-a-linked-list/problem

지난 번 DS 문제는 말단 노드에 새로운 노드를 삽입하는 문제였다면 이번 문제는 head 노드에 새로운 노드를 삽입하는 문제이다.

이 경우가 말단에 삽입할 때보다 더 간단한 것 같다.

SinglyLinkedListNode* insertNodeAtHead(SinglyLinkedListNode* llist, int data) {
    SinglyLinkedListNode* node = malloc(sizeof(SinglyLinkedListNode));

    node->data = data;
    node->next = NULL;
    if(llist == NULL)   llist = node;
    else{
        node->next = llist;
        llist = node;
    }
    return llist;

}

△ 작성한 부분

SMALL