Leetcode(链表)相交链表(c++)

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

例如,下面的两个链表:

A: a1 → a2

c1 → c2 → c3

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

注意:

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

解题思路:将两个链表右对齐,分别计算两个链表的长度,然后删除较长链表的前端结点,然后开始依次比较两个链表的结点是否相等,有则返回该结点,没有则返回NULL。

/**
 * 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) {
        if(!headA || !headB) return NULL;
        
        int Alen=0;
        int Blen=0;
        
        ListNode* A_first = headA;
        ListNode* B_first = headB;
        
        while(A_first){
            A_first=A_first->next;
            Alen++;
        }
        while(B_first){
            B_first=B_first->next;
            Blen++;
        }
        
        if(Alen>Blen){
            for(int i=0;i<(Alen-Blen);++i)
                headA=headA->next;
        }
        if(Alen<Blen){
            for(int j=0;j<(Blen-Alen);++j)
                headB=headB->next;
        }
        
        while(headA){
            if(headA==headB) return headA;
            headA=headA->next;
            headB=headB->next;
        }
        return NULL;
    }
};

你可能感兴趣的:(LeetCode)