[Linked List]링크드 리스트 실습2 : node mid add 함수 생성 (=insert)

이전 실습에 이어 num번째 노드 뒤 위치에 data 라는 노드를 추가하도록 만들어봅니다. (=insert)

 

이전 실습 링크는 아래를 통해 확인할 수 있습니다.

https://growingsaja.tistory.com/257

 

아래 코드는 전체 코드입니다.

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

# Node 추가 연결 함수 선언
def add(node, data):
    print(node.data)
    data = Node(data)
    while node.next:
        node = node.next
        print(node.data)
    node.next = data
    print(node.next.data)
    print("==")

node1 = Node(1) # 시작하는 첫 노드
print("==")
print(node1.data)
print("==")

# 2부터 10까지를 추가
for index in range(2, 11):
    add(node1, index)

# num번째 뒤에 data가 들어가도록 생성
def mid_add(node, num, data):
    data = Node(data)
    print(node.data)
    for index in range(num-1):
        node = node.next
        print(node.data)
    next_old = node.next
    node.next = data
    node = node.next
    print(node.data)
    node.next = next_old
    while node.next:
        node = node.next
        print(node.data)

print("====")

# 첫 노드가 node1인 3번째 뒤에 3.5가 들어가도록 생성
mid_add(node1, 3, 3.5)

위 사진은 이전 실습에서 추가된 내용입니다.