LeetCode 21. Merge Two Sorted Lists (Python3)

 

안녕하세요. 죠쵸입니다.

오늘 문제는 LeetCode 21번 문제입니다. 난이도는 "EASY"입니다. 데이터의 자료구조에는 크게 선형자료구조와 비선형자료구조로 나눌 수 있습니다. 이번 문제는 여러가지 선형 자료구조 중에 하나인 Linked List를 이용하는 문제입니다.

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

Merge Two Sorted Lists - LeetCode

 

#LeetCode 21. Merge Two Sorted Lists(Desctiption)

 

두개의 정렬된 정수 리스트(Linked List)를 입력받아, 두 리스트를 이용하여 하나의 정렬된 리스트(Linked List)를 반환하는 문제입니다.

 

 

#LeetCode 21. Merge Two Sorted Lists(Solution)

 

Linked List 는 연결된 Node 의 구조의 모양을 가집니다. 문제의 풀이를 간단히 설명히 드리면 두 리스트를 계속적으로 비교하면서 순차적으로 새로운 리스트에 추가하는 방법입니다. 새로운 노드를 생성하지 않고 기존의 노드를 재연결하는 방법으로 소스를 개선할 수도 있습니다. 하지만 오늘은 여기까지... ^^;;

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:
        root = current = ListNode()

        while l1 or l2:

            if (l1 and l2):
                if l1.val < l2.val:
                    value = l1.val
                    l1 = l1.next
                else:
                    value = l2.val
                    l2 = l2.next
            elif l1:
                value = l1.val
                l1 = l1.next
            elif l2:
                value = l2.val
                l2 = l2.next

            current.next = ListNode(value)
            current = current.next

        return root.next

 

그대신 동일한 풀이 방법이지만 , 재귀호출을 이용한 방법을 아래와 같이 공유 합니다. 종이에 문제 풀이를 적어가며 복귀하면 이해하시는데 도움이 되실 것 같습니다.

 

    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 = self.mergeTwoLists(l1.next, l2)

        return l1

 

방문해 주셔서 감사합니다. 여러분의 공감하기(♥), 댓글과 구독은 저에게 힘이 됩니다.

이상으로 죠쵸였습니다.