链表组件
给定一个链表(链表结点包含一个整型值)的头结点 head
。
同时给定列表 G
,该列表是上述链表中整型值的一个子集。
返回列表 G
中组件的个数,这里对组件的定义为:链表中一段最长连续结点的值(该值必须在列表 G
中)构成的集合。
示例 1:
输入:
head: 0->1->2->3
G = [0, 1, 3]
输出: 2
解释:
链表中,0 和 1 是相连接的,且 G 中不包含 2,所以 [0, 1] 是 G 的一个组件,同理 [3] 也是一个组件,故返回 2。
示例 2:
输入: head: 0->1->2->3->4 G = [0, 3, 1, 4] 输出: 2 解释: 链表中,0 和 1 是相连接的,3 和 4 是相连接的,所以 [0, 1] 和 [3, 4] 是两个组件,故返回 2。
思想:将数组加入list集合中,外层循环定义cur指向头结点,内层循环定义sign标记cur,再判断链表结点元素是否在集合中,在集合中继续下一个,不相等直接返回,
判断当前cur是否等于sign,不等于,计数器加1;
代码如下:
public int numComponents(ListNode head, int[] arr) {
List<Integer> list = new ArrayList<Integer>();
for(int i:arr){
list.add(i);
}
ListNode cur = head;
int count = 0;
while(cur!=null){
ListNode sign = cur;
while(sign!=null){
if(list.contains(sign.val)){
sign = sign.next;
}else{
break;
}
}
if(sign!=cur){
count++;
cur = sign;
}else{
cur = cur.next;
}
}
return count;
}