【Lintcode】1371. Linked List Components

题目地址：

https://www.lintcode.com/problem/linked-list-components/description

给定一个链表，题目保证链表中数字彼此不同，再给定一个数组$G$；如果链表中的两个节点$a$和$b$，$a$在$b$之前并且从$a$到$b$的路径上所有节点都属于$G$（或者$b$在$a$之前并且从$b$到$a$的路径上所有节点都属于$G$），那么我们说$a$与$b$等价。容易验证这是个等价关系。问链表里有多少个等价类。

因为要快速查询某个数是否在$G$里，我们先将$G$的所有数加进一个哈希表。接着从链表头开始遍历，遇到在$G$中的数的时候记等价类个数加一，接着将指针不停向后移动直到走到不在$G$里的数为止，然后重复上述过程。代码如下：

import java.util.HashSet;
import java.util.Set;

public class Solution {
    /**
     * @param head: the head
     * @param G: an array
     * @return: the number of connected components in G
     */
    public int numComponents(ListNode head, int[] G) {
        // Write your code here
        int res = 0;
        Set<Integer> set = new HashSet<>();
        for (int i : G) {
            set.add(i);
        }
        
        ListNode dummy = new ListNode(0), prev = dummy;
        dummy.next = head;
        
        while (prev.next != null) {
        	// 走到了G里的数，记等价类个数res加一，
        	// 并把prev向后移动直到走到链表末尾或者走到了某个G外的数为止
            if (set.contains(prev.next.val)) {
                res++;
                while (prev.next != null && set.contains(prev.next.val)) {
                    prev = prev.next;
                }
            } else {
                prev = prev.next;
            }
        }
        
        return res;
    }
}

class ListNode {
    int val;
    ListNode next;
    ListNode(int x) {
        val = x;
    }
}

时间复杂度$O(n)$，空间$O(1)$。