[LeetCode] 2181번: 기존 연결 리스트 수정하기

Hint

1. 어떻게 하면 포인터 2개를 이용해서 기존 리스트를 원하는 결과로 수정할 수 있을까?
2. 포인터 1개는 수정하는 노드 위치에 (a), 다른 하나는 리스트 전체를 확인하는 용도(b)로 이용해보자!
3. 움직이는 포인터b가 그 다음 0에 도달할 때까지 숫자들을 수정하는 포인터a 위치에 더하면 된다.
4. 마지막 노드는 null에 연결되도록 바꿔야한다.

 

LeetCode 2181번: Merge Nodes in Between Zeros (풀어보기)

class Solution:
    def mergeNodes(self, head: Optional[ListNode]) -> Optional[ListNode]:
        # 포인터 2개 만들기
        # now = 덧셈할 포인터 위치
        # pointer = 리스트 전체를 확인할 포인터 위치
        now = pointer = head
        
        # pointer로 리스트 전체 확인하기
        while pointer.next:
            pointer = pointer.next
            
            # 0이 아닌 숫자면 now 위치에 숫자를 더해주기
            if pointer.val != 0:
                now.val += pointer.val
            
            # 0이면 now 위치 옮겨주기
            else:
                if pointer.next:
                    now = now.next
                    now.val = 0
                # pointer가 리스트 끝에 도달했다면 now의 다음 노드를 None으로 해주어 나머지 리스트 끊어내기
                else:
                    now.next = None
        
        # now 포인터로 변경한 head 리스트를 반환하기
        return head