面试题52-两个链表的第一个公共节点

题目：

输入两个链表，找出它们的第一个公共节点。链表节点定义如下：

struct ListNode
{
	int m_nValue;
	ListNode *m_pNext;
	ListNode(int val): m_nValue(val), m_pNext(NULL) {};
};

解题思路

我们采用空间复杂度为O（1）和时间复杂度为O（m+n）的方法：
如下图：

c1是两个链表的第一个公共节点。我们可以先遍历一次得到它们的长度分别为5和6，也就是较长的链表与较短的链表相比多一个节点。第二次先在长的链表上走1步，到达节点b2。接下来分别从节点a1和b2出发同时遍历两个节点，直到找到它们第一个相同的节点c1，这就是我们想要的结果。

C++实现

class Solution
{
public:
	ListNode*FindFirstCommonNode(ListNode *pHead1, ListNode*pHead2)
	{	
		if (!pHead1 || !pHead2)
			return nullptr;
		//得到两个链表的长度
		unsigned int nLength1 = GetListLength(pHead1);
		unsigned int nLength2 = GetListLength(pHead2);
		int nLengthDif = nLength1 - nLength2;
		ListNode*pListHeadLong = pHead1;
		ListNode*pListHeadShort = pHead2;
		if (nLength1 < nLength2)
		{
			pListHeadLong = pHead2;
			pListHeadShort = pHead1;
			nLengthDif = nLength2 - nLength1;
		}
		//先在链表上走几步，再同时在两个链表上遍历
		for (int i = 0; i < nLengthDif; ++i)
			pListHeadLong = pListHeadLong->m_pNext;
		while (pListHeadLong&&pListHeadShort&&pListHeadLong != pListHeadShort)
		{
			pListHeadLong = pListHeadLong->m_pNext;
			pListHeadShort = pListHeadShort->m_pNext;
		}
		return pListHeadLong;
	}
	unsigned int GetListLength(ListNode*pHead)
	{
		unsigned int nLength = 0;
		ListNode*pNode = pHead;
		while (pNode != nullptr)
		{
			++nLength;
			pNode = pNode->m_pNext;
		}
		return nLength;
	}
	//Leetcode上的代码
	//采用两个指针最多遍历两次链表的方式
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        if (headA == nullptr || headB == nullptr) {
            return nullptr;
        }
        ListNode *pA = headA, *pB = headB;
        while (pA != pB) {
            pA = pA == nullptr ? headB : pA->next;
            pB = pB == nullptr ? headA : pB->next;
        }
        return pA;
    }
};

测试用例

1. 功能测试（输入的两个链表有公共节点：第一个公共节点在链表的中间，第一个公共节点在链表的末尾；第一个公共节点是链表的头节点；输入的两个链表没有公共节点）
2. 特殊输入测试（输入的链表头节点是Nullptr指针）