[LeetCode] 21. Merge Two Sorted Lists [Python(파이썬), JAVA(자바)]

'파이썬 알고리즘 인터뷰'를 보고 작성한 글입니다. 😀

문제 👉 <Merge Two Sorted Lists - LeetCode>

1. 문제 (두 정렬 리스트 병합)

정렬되어 있는 두 연결리스트를 합쳐라.

2. 풀이

1. 반복문을 이용한 풀이
2. 재귀를 이용한 풀이

3. 코드 (Python)

• 반복문을 이용한 풀이

# 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:
        head = tmp = ListNode(None)

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

        if l1:
            tmp.next = l1
        elif l2:
            tmp.next = l2

        return head.next

• 재귀를 이용한 풀이

# 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:
            return l2
        if not l2:
            return l1

        if l1.val > l2.val:
            l1, l2 = l2, l1
        l1.next = self.mergeTwoLists(l1.next, l2)
        return l1

• 결과 :

방식	Status	Runtime	Memory	Language
반복문	[Accepted]	28 ms	14.1 MB	python3
재귀	[Accepted]	40 ms	14.4 MB	python3

4. 코드 (Java)

• 반복문을 이용한 풀이

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {        
        ListNode ret = new ListNode();
        ListNode curr = ret;

        while (l1 != null && l2 != null) {
            if (l1.val < l2.val) {
                curr.next = l1;
                l1 = l1.next;
            } else {
                curr.next = l2;
                l2 = l2.next;
            }
            curr = curr.next;
        }

        if (l1 != null) curr.next = l1;
        else if (l2 != null) curr.next = l2;

        return ret.next;
    }
}

• 재귀를 이용한 풀이

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        // 종료 조건
        if (l1 == null) return l2;
        if (l2 == null) return l1;

        if (l1.val > l2.val) {
            ListNode tmp = l1;
            l1 = l2;
            l2 = tmp;
        }

        l1.next = mergeTwoLists(l1.next, l2);

        return l1;
    }
}

• 결과 :

방식	Status	Runtime	Memory	Language
반복문	[Accepted]	0 ms	38.6 MB	java
재귀	[Accepted]	0 ms	38.4 MB	java

---

References

---

🏋🏻 개인적으로 공부한 내용을 기록하고 있습니다.
잘못된 부분이 있다면 과감하게 지적해주세요!! 🏋