[LeetCode] 1721번 문제를 풀어보았다. (ft. java)

2022년 04월 04일 월요일 - 왜... 월요일...?

오늘 올려볼 문제는 1721번 Swapping Nodes in a Linked List 이라는 문제이다.

사진을 클릭하면 해당 문제로 이동합니다.

오늘도 LeetCode 사이트 오늘의 문제를 가지고 왔다.

아니 Value만 바꿔도 되는거 왜 안알려줬어!!!!

입력

사진에서도 볼 수 있듯이 Linked List 1개와 int값 1개가 입력으로 들어온다.

---

풀이 및 코드

앞에서 k번째 node와 뒤에서 k번째 node를 바꾸어서 리턴하는 문제다.

오늘은 시행착오를 좀 겪었다.

정석적으로 바꾸려는 node들의 앞뒤를 저장해서 이어주고 떼주고... 하다가 너무 힘들어서 HashMap을 사용해서 indexing을 하고 문제를 풀었다.

이제 코드를 봐보자!

풀이코드

class Solution {
    public ListNode swapNodes(ListNode head, int k) {
        HashMap<Integer, ListNode> map = new HashMap<>();
        int length = 0;
        ListNode start = head, end = head;

        for(;start != null; start = start.next)
        {
            length++;
            map.put(length, start);
        }

        if(length == 1) return head;

        if(k > length / 2) k = length - k + 1;

        if(length == 2)
        {
            map.get(1).next = null;
            map.get(2).next = map.get(1);

            return map.get(2);
        }
        else if(k == 1)
        {
            map.get(1).next = null;
            map.get(length).next = map.get(2);
            map.get(length - 1).next = map.get(1);

            return map.get(length);
        }
        else if(2 * k == length)
        {
            map.get(k - 1).next = map.get(k + 1);
            map.get(k + 1).next = map.get(k);
            map.get(k).next = map.get(k + 2);

            return head;
        }
        else
        {
            map.get(k - 1).next = map.get(length - k + 1);
            map.get(length - k + 1).next = map.get(k + 1);
            map.get(length - k).next = map.get(k);
            map.get(k).next = map.get(length - k + 2);

            return head;
        }
    }
}

---

제출 화면

사실 value만 바꿔도 되는줄 알았으면 훨씬 쉽게 문제를 풀었을 것이다.

내일은 좀 꼼수를 부리더라도 편하게 풀어야겠다.

내일도 문제를 풀어서 블로그에 글을 쓸 수 있으면 좋겠다.