리트코드 - 21. Merge Two Sorted Lists

아직 연결리스트 자료구조가 머릿속에서 정리가 잘 안된다. 

 

문제

leetcode.com/problems/merge-two-sorted-lists/

Merge Two Sorted Lists - LeetCode

 

코드

1) 순차적으로 붙여넣기

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
        result = ListNode()
        current = result

        while l1 and l2:
            if l1.val < l2.val:
                current.next = l1
                l1 = l1.next
            else:
                current.next = l2
                l2 = l2.next
            current = current.next

        #남은 노드 처리
        if l1:
            current.next = l1

        elif l2:
            current.next = l2

        return result.next

 

다른 코드

1) 재귀 구조로 연결

이 문제는 정렬된 리스트가 주어지므로, 병합 정렬에서 마지막 조합 시 첫 번째 값부터 차례대로만 비교하면 한 번에 해결되듯이 여기에서도 이 방식으로 풀면된다는데,, 

아직 이해가 되지는 않는다. 계속 보면서 이해하기 위해 작성 해놔야겠다.

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
      if (not l1) or (l2 and l1.val > l2.val):
          l1, l2 = l2, l1

      if l1:
          l1.next = mergeTwoLists(l1.next, l2)

      return l1