题目
示例答案
答案
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
struct ListNode* curA = headA;
struct ListNode* curB = headB;
while(curA != curB){
curA = curA ?curA->next : headB;
curB = curB ? curB->next : headA;
}
return curA;
}
作者:nkion 链接:https://leetcode.cn/problems/intersection-of-two-linked-lists/solutions/2477297/cyu-yan-10xing-dai-ma-by-jian-ting-yuan-59sq/ 来源:力扣(LeetCode)著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
理解
代码的主要思想是使用两个指针分别在两个链表上移动,初始时curA指向headA,curB指向headB。while循环的条件是检查curA和curB是否相等,如果相等,则找到了交点,如果不相等则继续循环。
代码的巧妙之处在于while循环内部的内容,如果curA不为NULL,则继续向后移动,如果curA为NULL,那么就将curA指向headB,这样curA遍历完链表headA后,从headB开始遍历;对于curB同理,遍历完headB然后遍历headA。(可以使用上面的相交链表实验一下)
当两个链表长度相等的时候,curA和curB会在交点相遇,当两个链表长度不相等时,由于curA和curB都会遍历两个链表,所有会在第二次遍历的时候在交点相遇
对于循环中的三元运算符的表达,如果不好理解可以换成if语句
struct ListNode *getIntersectionNode(struct ListNode *headA, struct ListNode *headB) {
struct ListNode *curA = headA;
struct ListNode *curB = headB;
while (curA != curB) {
if (curA) {
curA = curA->next;
} else {
curA = headB;
}
if (curB) {
curB = curB->next;
} else {
curB = headA;
}
}
return curA;
}