insertNodeAtPosition.py

"""
You’re given the pointer to the head node of a linked list, an integer to add to the list and the position at which the integer must be inserted. Create a new node with the given integer, insert this node at the desired position and return the head node.

A position of 0 indicates head, a position of 1 indicates one node away from the head and so on. The head pointer given may be null meaning that the initial list is empty.

As an example, if your list starts as
and you want to insert a node at position with , your new list should be

Function Description Complete the function insertNodeAtPosition in the editor below. It must return a reference to the head node of your finished list.

insertNodeAtPosition has the following parameters:

    head: a SinglyLinkedListNode pointer to the head of the list
    data: an integer value to insert as data in your new node
    position: an integer position to insert the new node, zero based indexing

Input Format

The first line contains an integer
, the number of elements in the linked list.
Each of the next lines contains an integer SinglyLinkedListNode[i].data.
The last line contains an integer

.

Constraints

, where is the
element of the linked list.

    .

Output Format

Return a reference to the list head. Locked code prints the list for you.

Sample Input

3
16
13
7
1
2

Sample Output

16 13 1 7

Explanation

The initial linked list is 16 13 7. We have to insert
at the position which currently has in it. The updated linked list will be 16 13 1 7
"""

#!/bin/python3

import math
import os
import random
import re
import sys

class SinglyLinkedListNode:
    def __init__(self, node_data):
        self.data = node_data
        self.next = None

class SinglyLinkedList:
    def __init__(self):
        self.head = None
        self.tail = None

    def insert_node(self, node_data):
        node = SinglyLinkedListNode(node_data)

        if not self.head:
            self.head = node
        else:
            self.tail.next = node


        self.tail = node

def print_singly_linked_list(node, sep, fptr):
    while node:
        fptr.write(str(node.data))

        node = node.next

        if node:
            fptr.write(sep)

# Complete the insertNodeAtPosition function below.

#
# For your reference:
#
# SinglyLinkedListNode:
#     int data
#     SinglyLinkedListNode next
#
#
def insertNodeAtPosition(head, data, position):

    if head == None:
        head = SinglyLinkedListNode(data)

    else:
        temp = head
        last = SinglyLinkedListNode(data)

        if position == 0:
            last.next = temp
            head = last

        else:
            count = 1

            while count != position:
                temp = temp.next
                count += 1

            last.next = temp.next
            temp.next = last

        return head
        

    return head

if __name__ == '__main__':
    # fptr = open(os.environ['OUTPUT_PATH'], 'w')

    # llist_count = int(input())

    # llist = SinglyLinkedList()

    # for _ in range(llist_count):
    #     llist_item = int(input())
    #     llist.insert_node(llist_item)

    # data = int(input())

    # position = int(input())

    # llist_head = insertNodeAtPosition(llist.head, data, position)

    # print_singly_linked_list(llist_head, ' ', fptr)
    # fptr.write('\n')

    # fptr.close()

    llist_count = 3

    llist = SinglyLinkedList()

    array = [16, 13, 7]

    for _ in range(llist_count):
        llist_item = array[_]

        print(llist_item)

        llist.insert_node(llist_item)

    data = 1

    position = 2

    llist_head = insertNodeAtPosition(llist.head, data, position)

    result = []

    while llist_head:
        result.append(llist_head.data)

        llist_head = llist_head.next

    print(result)