[Java] 21. Merge Two Sorted Lists

문제 파악

정렬된 두 연결 리스트의 첫번째 노드를 입력받는다.

 

두 연결 리스트를 하나의 정렬된 연결리스트로 만들어야한다.

 

반드시 입력받은 노드로 구성해야한다.

 

---

풀이

 

1️⃣

💡 떠오른 Idea

각 리스트의 값을 비교하고, 작은 값을 포함하는 노드를 추가하는 식으로 합치면 된다.

단, 비교할 노드가 없어진 경우 남아있는 노드를 순차적으로 추가해주면 된다.

 

 

class Solution {
    public ListNode mergeTwoLists(ListNode list1, ListNode list2) {
        ListNode head = new ListNode();
        ListNode ans = head;
        ListNode result;

        while (list1 != null || list2 != null) {
            if (list1 != null && list2 != null) {
                if (list1.val >= list2.val) {
                    result = list2;
                    list2 = list2.next;
                } else {
                    result = list1;
                    list1 = list1.next;
                }
            } else if (list1 == null) {
                result = list2;
                list2 = list2.next;
            } else {
                result = list1;
                list1 = list1.next;
            }

            ans.next = result;
            ans = ans.next;
        }
        return head.next;
    }
}

 

 

결과

 

---

📖 회고

 

이전에 풀었던

 

Add Two Numbers - LeetCode

 

이 문제의 쉬운 버전 문제로 느껴져서 금방 풀이할 수 있었다.

 

이 문제를 먼저 풀어봤다면, 더 좋지 않았을까? 생각도 들었다. ㅎㅎ