30-InSange

[InSange]

🔗 문제 링크

Delete Nodes From Linked List Present in Array

✔️ 소요된 시간

30분

✨ 수도 코드

문제 개요

정수형 연결리스트 head와 정수 배열 nums가 매개 변수로 주어졌을 때 nums안에 속한 노드들을 뺀 연결리스트를 출력

첫 번째 풀이 방식

문제를 잘 못 이해하였다. 그러나 큰 틀은 달라진게 없어서 금방 고칠 수 있었음.
처음에 nums에 속하지 않는 값을 지닌 노드들을 제거하는 줄 알았다.

그러나 문제를 다시 읽었을 때 반대였다는 것을 깨달음.

1. nums에 속하는 값들을 맵에 추가하여 분류해준다. (um에 존재하는 값 = 빼야하는 값)
2. 현재 노드가 um에 속한 값일 경우 removeList의 next 노드 값을 현재 head값으로 세팅해주고 업데이트 해준다.
3. 만약 removeList가 nullptr일 경우 현재 head를 세팅.
4. 그렇지 않을 경우. 다음 노드를 탐색.

class Solution {
public:
    ListNode* modifiedList(vector<int>& nums, ListNode* head) {
        ListNode* removeList = nullptr;
        ListNode* removeHead = nullptr;
        unordered_map<int, bool> um;
        
        for(int& val : nums) um[val] = true;
        
        while(head != nullptr)
        {
            cout << head->val << " ";
            if(um[head->val] == false)
            {
                if(removeList) 
                {
                    cout << "removeHead = " << removeList->val << " ";
                    removeList->next = head;
                    removeList = removeList->next;
                }
                else 
                {
                    removeList = head;
                    removeHead = removeList;
                }
            }
            head = head->next;
            if(removeList) removeList->next = nullptr;
        }
        
        return removeHead;
    }
};

처음에 애를 먹었던게

 removeList = removeList->next;

removeList에 값을 넣어주게 되면 다음 값으로 갱신시켜줘야 된다.
갱신 시켜주지 않을 경우 맨 처음 들어온 head값과 head->next 값 두 개만 탐색을 하기 때문이다.
그렇다고 마지막에 head값을 갱신해줄 때 해당 코드가 들어가게 되면 무한 루프가 된다.

그럼 새로운 노드를 만들어서 하면 되지 않느냐. 그럼 그만큼 메모리가 차지하는 비중이 더 생기기 때문에 현재 주어진 연결 리스트 노드만 써서 활용해보고 싶었다가 해당 코드가 나오게 되었다.

두번째 풀이

문제를 풀고 생각해보다가 어차피 연결리스트 노드만 사용해서 하는 것이고 um에 해당하는, nums 번호에 속할 경우 해당 노드를 그냥 넘겨버리면 되지 않을까? 생각이 떠올랐다.

기존 코드를 재활용하여

removeList의 head를 빈 노드 값으로 채워주었다.
이 부분은 추후 반복문에서 비교할 때 조건문을 현재노드 -> next를 기준으로 해주기 위해서 빈 노드를 추가 해주었다.

removeList의 next를 연결리스트 head로 설정한 후 반복문을 실행한다.

1. 현재노드 -> next의 값이 존재할 경우. 해당 next의 값이 nums에 속할 경우. 아예 다음 노드를 현재 노드와 연결시켜준다.
2. 그렇지 않을 경우 다음 노드를 그냥 탐색

class Solution {
public:
    ListNode* modifiedList(vector<int>& nums, ListNode* head) {
        ListNode* removeList = new ListNode();
        ListNode* removeHead;
        unordered_map<int, bool> um;
        
        for(int& val : nums) um[val] = true;
        
        removeList->next = head;
        removeHead = removeList;

        while(removeHead->next)
        {
            if(um[removeHead->next->val] == true) removeHead->next = removeHead->next->next;
            else removeHead = removeHead->next;
        }
        
        return removeList->next;
    }
};

더 간결해진 것을 확인할 수 있으며

효율도 더 올라간 것을 확인할 수 있다.

단순하다.

📚 새롭게 알게된 내용

9월 11일 까지는 당분간 캡디 개발로 바빠서 힘들 것 같습니다..! 리뷰는 최대한 시간 짬낼때마다 하겠으나 잠시 PR은 미루겠습니다.