LeetCode刷题之no.160

关键知识

链表的处理

解题思路

一种比较简单的方法就是将以一条链表中的每个结点的跟另一条链表的每个结点相比较，首个相等的结点即为两个链表的交的起始结点。如果简单地双层遍历，会使得时间复杂到O(n^2)，如果用到stl中的set，可以将平均复杂度降到O(n)，但是最差情况复杂度仍为 到O(n^2)。
另一种方法就是两条链表相遇后就不会分开，则可以从后面开始数起最后不相同数量的结点，即从较短的链表的起始结点开始同步遍历，遍历过程中如果相遇则链表相交。

简单做法

遍历第一条链表，将每个结点加进set里面。然后遍历第二条链表，将每个结点和set的内容相比较，如果set里面存在该结点，则说明两条链表存在交集，交集起点为第一个查找成功的结点。如果查找结束仍未找到，则说明两条链表不存在交集。
时间复杂度： 平均为O(n)，最差为O(n^2)
空间复杂度： O(1)

最优做法

用两个指针分别记录两条链表的起始结点。首先计算两条链表各自长度，然后计算长度差，再排除多出来的那部分结点。然后开始两条链表同时开始遍历，如果遍历期间相遇，则说明两条存在交集，否则不存在。在第一次相遇的结点即为交集的起始结点。
时间复杂度： O(n)
空间复杂度： O(1)

题目

简答做法

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        set<ListNode*> sa;
        ListNode* p = headA;
        while (p) {
            sa.insert(p);
            p = p->next;
        }
        p = headB;
        while (sa.find(p) == sa.end() && p) p = p->next;
        return p;
    }
};

最优解法

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        set<ListNode*> sa;
        ListNode* p1 = headA;
        ListNode* p2 = headB;
        int n1 = 0, n2 = 0;  // 两条链表的长度
        while (p1) {
        	n1++;
        	p1 = p1->next;
        }
        while (p2) {
        	n2++;
        	p2 = p2->next;
        }
        int n = 0;
        p1 = headA;
        p2 = headB;
        if (n1 > n2) {
        	n = n1 - n2;
        	while (n>0) {
                p1 = p1->next;
                n--;
            }
        }
        if (n1 < n2) {
        	n = n2 - n1;
        	while (n>0) {
                p2 = p2->next;   
                n--;
            }
        }

        while (p1 && p2) {
        	if (p1 == p2) return p1;
        	else {
        		p1 = p1->next;
        		p2 = p2->next;
        	}
        }
        return nullptr;
    }
};

题目地址

Intersection of Two Linked Lists