알고리즘/이론정리
양방향 연결리스트(Doubly Linked Lists)
팽팽
2021. 9. 14. 19:01
양방향 연결리스트란?
한쪽으로만 링크를 연결하지 말고 양쪽으로 - 앞으로도(다음node) 뒤로도(이전node)진행가능
node의 구조 확장
class Node:
def __init__(self, item):
self.data = item
self.prev = None
self.next = None
리스트 처음과 끝에 dummy node를 두자
-> 데이터를 담고 있는node들은 모두 같은 모양
L.insertAfter
next = prev.next
newNode.prev <- prev
newNode.next <- next
next.prev <-newNode
리스트 마지막에 원소 삽입하는 경우는 어떻게?
getAt메서드를 개선하자!
포지션이 nodecount 절반보다 뒷쪽이면 tail기준으로 역방향으로 돌아오면 됨!
def getAt(self, pos):
if pos < 0 or pos > self.nodeCount:
return None
if pos > self.nodeCount // 2:
i = 0
curr = self.tail
while i < self.nodeCount - pos + 1:
curr = curr.prev
i += 1
else:
i = 0
curr = self.head
while i < pos:
curr = curr.next
i += 1
return curr
즉 ,
지금까지는(단방향 연결리스트) tail이 마지막 노드를 가리키고 있다고 해도
앞쪽에서부터 하나하나 했기 때문에 시간이 리스트의 길이에 비례했다.
양방향은 마지막을 먼저가고 앞 노드를 찾아가는 방식이다.
- insertAfter메서드
def insertAfter(self, prev, newNode):
next = prev.next
newNode.prev = prev
newNode.next = next
prev.next = newNode
next.prev = newNode
self.nodeCount += 1
return True
양방향 연결리스트 전체 코드
class Node:
def __init__(self, item):
self.data = item
self.prev = None
self.next = None
class DoublyLinkedList:
def __init__(self):
self.nodeCount = 0
self.head = Node(None)
self.tail = Node(None)
self.head.prev = None
self.head.next = self.tail
self.tail.prev = self.head
self.tail.next = None
def __repr__(self):
if self.nodeCount == 0:
return 'LinkedList: empty'
s = ''
curr = self.head
while curr.next.next:
curr = curr.next
s += repr(curr.data)
if curr.next.next is not None:
s += ' -> '
return s
def getLength(self):
return self.nodeCount
def traverse(self):
result = []
curr = self.head
while curr.next.next:
curr = curr.next
result.append(curr.data)
return result
def getAt(self, pos):
if pos < 0 or pos > self.nodeCount:
return None
if pos > self.nodeCount // 2:
i = 0
curr = self.tail
while i < self.nodeCount - pos + 1:
curr = curr.prev
i += 1
else:
i = 0
curr = self.head
while i < pos:
curr = curr.next
i += 1
return curr
def insertAfter(self, prev, newNode):
next = prev.next
newNode.prev = prev
newNode.next = next
prev.next = newNode
next.prev = newNode
self.nodeCount += 1
return True
def insertAt(self, pos, newNode):
if pos < 1 or pos > self.nodeCount + 1:
return False
prev = self.getAt(pos - 1)
return self.insertAfter(prev, newNode)
연습문제 1 : 양방향 연결 리스트 역방향 순회
def reverse(self):
result =[]
curr = self.tail
while curr.prev.prev:
curr = curr.prev
result.append(curr.data)
return result
정방향 순회의 반대로 tail에서 시작해주면 됨
연습문제 2 : 양방향 연결 리스트 노드 삽입
def insertBefore(self, next, newNode):
prev =next.prev
newNode.prev = prev
newNode.next = next
prev.next = newNode
next.prev = newNode
self.nodeCount += 1
return True
inserAfter와 똑같은 알고리즘. prev만 재정의해주면 됨