-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdeletenode at pos.py
91 lines (62 loc) · 2.01 KB
/
deletenode at pos.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# Delete the node at a given position in a linked list and return a reference to the head node. The head is at position 0. The list may be empty after you delete the node. In that case, return a null value.
# Example
# After removing the node at position , .
# Function Description
# Complete the deleteNode function in the editor below.
# deleteNode has the following parameters:
# - SinglyLinkedListNode pointer llist: a reference to the head node in the list
# - int position: the position of the node to remove
# Returns
# - SinglyLinkedListNode pointer: a reference to the head of the modified list
# 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)
def deleteNode(llist, position):
# Write your code here
head = llist
cur = head
if not head:
return None
else:
c = 0
while c!=position:
pre = cur
cur = cur.next
c += 1
pre.next = cur.next
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)
# position = int(input())
# llist1 = deleteNode(llist.head, position)
# print_singly_linked_list(llist1, ' ', fptr)
# fptr.write('\n')
# fptr.close()