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

2022년 03월 11일 금요일 - T1 15 연승!!

오늘 올려볼 문제는 61번 Rotate List 이라는 문제이다.

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

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

오늘 문제 꽤 재밌었음!

입력

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

---

풀이 및 코드

k 만큼 Linked List를 돌려서 리턴하는 문제다.

오늘은 처음부터 정답을 생각해냈다.

일단 리스트의 끝을 리스트의 처음과 연결하고 리스트의 길이를 구한다.

그리고 리스트 길이 - (k % 리스트 길이) 만큼 노드를 이동시키고 리스트를 끊으면 리스트를 돌린 것과 같은 효과가 난다.

위와 같은 방식으로 해당 문제를 풀었다.

이제 코드를 봐보자!

풀이코드

class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        if(head == null) return head;

        ListNode node = head;
        int index;

        for(index = 1; node.next != null; index++) node = node.next;
        node.next = head;

        for(int i = 0; i < index - (k % index); i++) node = node.next;
        head = node.next;
        node.next = null;

        return head;
    }
}

---

제출 화면

오늘 문제는 꽤나 재밌게 풀었다.

내일도 이런 문제가 나왔으면 좋겠다!

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