#include<stdio.h>
#include<stdlib.h>

//Create a basic structure for NODE from which new nodes can be created.
struct node
{
    int data;
    struct node *link;
};

//Initialize 3 pointers as globals so that they do not need to be passed in functions.
struct node *header, *ptr, *temp;

//Prototypes for various user defined functions.
void insert_front();
void insert_end();
void display();


void insert_front()
{
    int data_value;

    printf("\nEnter data of the node: ");
    scanf("%d", &data_value);

    temp = (struct node *) malloc(sizeof(struct node));

    temp->data = data_value;
    temp->link = header->link;
    header->link = temp;
}


void insert_end()
{
    int data_value;

    printf("\nEnter data of the node: ");
    scanf("%d", &data_value);

    temp = (struct node *) malloc(sizeof(struct node));

    //Traverse to the end of the linked list.
    ptr = header;
    while(ptr->link != NULL)
    {
        ptr = ptr->link;
    }

    temp->data = data_value;
    temp->link = ptr->link;
    ptr->link = temp;
}
void display()
{
    printf("\nContents of the linked list are: \n");
    //Print the contents of the linked list starting from header
    ptr = header;
    while(ptr->link != NULL)
    {
        ptr = ptr->link;
        printf("%d ", ptr->data);
    }
}






int main()
{
    int choice;
    int cont = 1;

    //Allocate memory for header node.
    header = (struct node *) malloc(sizeof(struct node));



    //Set the content of header node
    header->data = NULL;
    header->link = NULL;

    while(cont == 1)
    {
        //Display menu to the user
        printf("\n1. Insert at front\n");
        printf("\n2. Insert at end\n");
        printf("\n3. Display linked list\n");
        printf("\nEnter your choice: ");
        scanf("%d", &choice);

        switch(choice)
        {
            case 1:
                insert_front();
                break;
            case 2:
                insert_end();
                break;

            case 3:
                display();
                break;
        }

        printf("\n\nDo you want to continue? (1 / 0): ");
        scanf("%d", &cont);
    }

    return 0;
}