剑指Offer_面试题37_两个链表的第一个公共结点

题目描述

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

解法一：允许使用辅助空间，利用哈希表std::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 = NULL;
        unordered_set<ListNode*> hash;
        while(pHead1)
        {
            hash.insert(pHead1);
            pHead1 = pHead1->next;
        }
        while(pHead2)
        {
            if(hash.find(pHead2) != hash.end())
            {
                res = pHead2;
                break;
            }
            pHead2 = pHead2->next;
        }
        return res;
    }
};

解法二：辅助空间栈，两个链表分别入栈，然后出栈到第一个不相同的结点，前一个就是第一个公共结点。

class Solution {
public:
    ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
        stack<ListNode*> stack1;
        stack<ListNode*> stack2;
        ListNode *p1 = pHead1;
        ListNode *p2 = pHead2;
        ListNode *last = NULL;
        while(p1)
        {
            stack1.push(p1);
            p1 = p1->next;
        }
        while(p2)
        {
            stack2.push(p2);
            p2 = p2->next;
        }
        while(!stack1.empty() && !stack2.empty())
        {
            p1 = stack1.top();
            p2 = stack2.top();
            if(p1 != p2){
                break;
            }
            else
                last = p1;
            stack1.pop();
            stack2.pop();
        }
        return last;
    }
};

解法三：不能使用辅助空间，分别遍历两个链表得到长度，长的先走长度差步，然后一起走，找到第一个相同结点即为第一个公共结点。

class Solution {
public:
    ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
        int len1 = 0, len2 = 0;
        ListNode *p1 = pHead1, *p2 = pHead2;
        if(!p1 || !p2) return NULL;
        while(p1){
            ++len1;
            p1 = p1->next;
        }
        while(p2){
            ++len2;
            p2 = p2->next;
        }
        int step = 0;
        if(len1 < len2){
            step = len2 - len1;
            p1 = pHead2;
            p2 = pHead1;
        }
        else{
            step = len1 - len2;
            p1 = pHead1;
            p2 = pHead2;
        }
       for(int i = 0; i < step; ++i)
           p1 = p1->next;
        while(p1 && p2 && p1 != p2){
            p1 = p1->next;
            p2 = p2->next;
        }
        if(p1 == p2)
            return p1;
        return NULL;
    }
};