36.两个链表的第一个公共节点

题目描述

输入两个链表，找出它们的第一个公共结点。

思路：用一个unordered_set记录已访问的链表的地址，第一个重复出现的地址，即为第一个公共结点

代码：

/*
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};*/
class Solution {
public:
    ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
        //ListNode* res;
        ListNode* p1 = pHead1;
        ListNode* p2 = pHead2;
        unordered_set<ListNode*> s;
        while(p1 != NULL || p2 != NULL) {
            if(p1) {
                if(s.find(p1) != s.end()) {
                    return p1;
                }
                s.insert(p1);
                p1 = p1->next;
            }
            
            if(p2) {
                if(s.find(p2) != s.end()) {
                    return p2;
                }
                s.insert(p2);
                p2 = p2->next;
            }
             
        }
        return NULL;
    }
};

思路二：

先求出两个链表的总长度，让较长链表的头指针移动两链表长度差，此时再让两个头指针同时往后移动，则一定会同时到达第一个公共结点，每次移动后判断两个头指针是否相同即可，若相同，则此时的结点即为第一个公共结点。

该方法相比于第一种，节省了额外的空间开销，两种方法的时间复杂度均是O(M+N)

代码：

/*
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};*/
class Solution {
public:
    ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
        int len1 = getLen(pHead1);
        int len2 = getLen(pHead2);
        int tmp = len1 - len2;
        
        if(len1 > len2) {
            while(tmp--) {
                pHead1 = pHead1->next;
            }
        } else {
            tmp = -tmp;
            while(tmp--) {
                pHead2 = pHead2->next;
            }
        }
        while(pHead1 != NULL && pHead2 != NULL && pHead1 != pHead2) {
            pHead1 = pHead1->next;
            pHead2 = pHead2->next;
        }
        return pHead1;
    }
    
    int getLen(ListNode* p) {
        int len = 0;
        while(p) {
            ++len;
            p = p->next;
        }
        return len;
    }
};