最近在学习链表时遇到的题目,抄过来记录下,防止以后忘记。
题目:给定两个可能有环也可能无环的单链表,头节点head1和head2.请实现一个函数,如果两个链表相交,请返回相交的第一个节点。如果不相交,返回null;
要求:如果两个链表长度之和为N,时间复杂度请达到O(N),额外空间复杂度请达到O(1)
public static Node getIntersectNode(Node head1,Node head2) {
if(head1 == null || head2 == null) {
return null;
}
Node loop1 = getLoopNode(head1);
Node loop2 = getLoopNode(head2);
if(loop1 == null && loop2 == null) {
return noLoop(head1, head2);
}
if(loop1 != null && loop2 != null) {
return bothLoop(head1, loop1, head2, loop2);
}
return null;
}
//找到链表第一个入环节点,如果无环,返回null
public static Node getLoopNode(Node head) {
if(head == null || head.next == null || head.next.next == null) { //快指针每次走两步,慢指针每次走一步
return null;
}
Node n1 = head.next;
Node n2 = head.next.next;
while(n1 != n2) { //从循环中跳出去则说明就是在环上某个位置相遇
if(n2.next == null || n2.next.next == null) { //快指针提前走到结尾说明没有环
return null; //返回第一个入环节点是空
}
n2 = n2.next.next; //快指针每次走两步
n1 = n1.next; //慢指针每次走一步
}
n2 = head; //相遇后快指针回到开头,慢指针留在原地
while(n1 != n2) { //while不相遇时继续,相遇时停止
n1 = n1.next; //变成两个节点都是每次走一步
n2 = n2.next;
}
return n1;
}
//如果两个链表都无环,返回第一个相交节点,如果不相交则返回null
public static Node noLoop(Node head1,Node head2) {
if(head1 == null || head2 == null) {
return null;
}
Node cur1 = head1;
Node cur2 = head2;
int n = 0;
while(cur1.next != null) { //链表1来到最后一个节点停
n++;
cur1 = cur1.next;
}
while(cur2.next != null) { //链表2来到最后一个节点停
n--;
cur2 = cur2.next;
}
if(cur1 != cur2) { //链表1的最后一个内存地址与链表2的最后一个内存地址不相等则一定是不相交的
return null;
}
//n:链表一长度的值减去链表二长度的值
cur1 = n > 0 ? head1 : head2; //谁长谁的头就变成cur1
cur2 = cur1 == head1 ? head2 : head1; //谁短,谁的头变成cur2;
n = Math.abs(n); //n取绝对值
while(n != 0) {
n--;
cur1 = cur1.next; //长链表先走差值步
}
while(cur1 != cur2) {
cur1 = cur1.next; //然后长链表和短链表一起走
cur2 = cur2.next;
}
return cur1; //再次相遇则是相交的点
}
//两个有环链表,返回第一个相交节点,如果不相交返回null
public static Node bothLoop(Node head1,Node loop1,Node head2,Node loop2) {
Node cur1 = null;
Node cur2 = null;
if(loop1 == loop2) {
cur1 = head1;
cur2 = head2;
int n = 0;
while(cur1 != loop1) {
n++;
cur1 = cur1.next;
}
while(cur2 != loop2) {
n--;
cur2 = cur2.next;
}
cur1 = n > 0 ? head1 : head2;
cur2 = cur1 == head1 ? head2 : head1;
n = Math.abs(n);
while(n != 0) {
n--;
cur1 = cur1.next;
}
while(cur1 != cur2) {
cur1 = cur1.next;
cur2 = cur2.next;
}
return cur1;
}else {
cur1 = loop1.next;
while(cur1 != loop1) {
if(cur1 == loop2) {
return loop1;
}
cur1 = cur1.next;
}
return null;
}
}