サンダーボルト

相手モンスターを全て破壊する。

LeetCode Study : 141. Linked List Cycle

問題

Given a linked list, determine if it has a cycle in it.

To represent a cycle in the given linked list, we use an integer pos which represents the position (0-indexed) in the linked list where tail connects to. If pos is -1, then there is no cycle in the linked list.

Example 1:

Input: head = [3,2,0,-4], pos = 1
Output: true
Explanation: There is a cycle in the linked list, where tail connects to the second node.

Example 2:

Input: head = [1,2], pos = 0
Output: true
Explanation: There is a cycle in the linked list, where tail connects to the first node.

https://leetcode.com/problems/linked-list-cycle/

自分の解答

public boolean hasCycle(ListNode head) {
    Set<ListNode> set = new HashSet<>();
    while (head != null) {
        if (set.contains(head)) {
            return true;
        } else {
            set.add(head);
        }
        head = head.next;
    }
    return false;
}

コード理解

別解

public boolean hasCycle(ListNode head) {
    if (head == null || head.next == null) {
        return false;
    }
    ListNode slow = head;
    ListNode fast = head.next;
    while (slow != fast) {
        if (fast == null || fast.next == null) {
            return false;
        }
        slow = slow.next;
        fast = fast.next.next;
    }
    return true;
}

Time :  O(n) / Space :  O(1)

これはすごい発想だなぁ。あるリストがループしてるかどうかを、2つのポインターを違う速度で走らせて、追いついたらループしてたことにすると。

自分のよりもメモリ使ってないし、多少わからなかったら読みにくさはあるけどまぁ優秀だよなぁ。これは思いつかないなぁ。もはや知ってるか知らんかの世界な気もする。

だからこそ覚えておくか。

今後のための考え方

  • ループしているかどうかは2つのポインタを違う速度で走らせて追いついたらループしてる!