Leetcode链表easy | 160. 相交链表

编写一个程序,找到两个单链表相交的起始节点。

例如,下面的两个链表:

A: a1 → a2

c1 → c2 → c3

B: b1 → b2 → b3
在节点 c1 开始相交。

注意:

如果两个链表没有交点,返回 null.
在返回结果后,两个链表仍须保持原有的结构。
可假定整个链表结构中没有循环。
程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        ListNode* i =headA;  //  定义动态指针
        ListNode* j =headB;
        
        int lenA=1,lenB=1;
        
        while(i){
            i=i->next;
            lenA++;
        }  // 遍历A链表,计算长度
        while(j){
            j=j->next;
            lenB++;
        } //  遍历B链表,计算长度
        
        if(i!=j)  //  最后的节点不是同一个,说明没相交,返回NULL
            return NULL;
        
        int gap = abs(lenA-lenB);  //  计算长度差
        
        i=headA,j=headB;  //  动态指针归位
        if(lenA>lenB){
            while(gap){
                i=i->next;
                gap--;
            }
        }  //  若A链表较长,i指针先走gap步
        else{
            while(gap){
                j=j->next;
                gap--;
            }
        } // 若B链表较长,j指针先走gao步
        
        while(i!=j){
            i=i->next;
            j=j->next;
        }  //  循环直到i,j指针相遇
        
        return i;    //  返回相交的节点
        
    }
};

你可能感兴趣的:(Leetcode链表easy | 160. 相交链表)