매일 조금씩

Leet code (Easy) : 21. Merge Two Sorted Lists - JAVA 본문

알고리즘/LinkedList

Leet code (Easy) : 21. Merge Two Sorted Lists - JAVA

mezo 2024. 10. 16. 22:07
728x90
반응형

 

LinkedList 의 개념을 다시 잡을 수 있었다.

연결된 노드 객체들은 다음 노드를 가리키는 포인터를 갖는다.

사이클이 없다면, LinkedList의 헤드를 가리키는 더미 노드를 임의로 생성해서 활용해도 된다.

이 문제에선 그 더미노드를 사용하는 것이 핵심이다.

 

/**
 * 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 list1, ListNode list2) {
        // 헤드를 리턴하기 위해 헤드 앞에오는. 헤드를 가리키는 더미 노드를 생성.
        ListNode dummy = new ListNode();
        // 결과 링크드리스트를 만들기 위한 포인터를 더미를 가르키도록 초기화하면서 만듦.
        ListNode curr = dummy;

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

        // 남은 노드를 연결
        cur.next = (list1 != null)? list1 : list2;

        // 헤드 노드를 리턴
        return dummy.next;
    }
}

 

728x90
반응형