160. 相交鏈表(Python)

題目

難度:★★☆☆☆
類型:鏈表

編寫一個程序,找到兩個單鏈表相交的起始節(jié)點(diǎn)。

如下面的兩個鏈表:

示例

在節(jié)點(diǎn) c1 開始相交。

示例 1:

示例1

輸入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
輸出:Reference of the node with value = 8
輸入解釋:相交節(jié)點(diǎn)的值為 8 (注意,如果兩個列表相交則不能為 0)。從各自的表頭開始算起,鏈表 A 為 [4,1,8,4,5],鏈表 B 為 [5,0,1,8,4,5]。在 A 中,相交節(jié)點(diǎn)前有 2 個節(jié)點(diǎn);在 B 中,相交節(jié)點(diǎn)前有 3 個節(jié)點(diǎn)。

示例2:

示例2

輸入:intersectVal = 2, listA = [0,9,1,2,4], listB = [3,2,4], skipA = 3, skipB = 1
輸出:Reference of the node with value = 2
輸入解釋:相交節(jié)點(diǎn)的值為 2 (注意,如果兩個列表相交則不能為 0)。從各自的表頭開始算起,鏈表 A 為 [0,9,1,2,4],鏈表 B 為 [3,2,4]。在 A 中,相交節(jié)點(diǎn)前有 3 個節(jié)點(diǎn);在 B 中,相交節(jié)點(diǎn)前有 1 個節(jié)點(diǎn)。

示例3:

示例3

輸入:intersectVal = 0, listA = [2,6,4], listB = [1,5], skipA = 3, skipB = 2
輸出:null
輸入解釋:從各自的表頭開始算起,鏈表 A 為 [2,6,4],鏈表 B 為 [1,5]。由于這兩個鏈表不相交,所以 intersectVal 必須為 0,而 skipA 和 skipB 可以是任意值。
解釋:這兩個鏈表不相交,因此返回 null。

注意:

如果兩個鏈表沒有交點(diǎn),返回 null.
在返回結(jié)果后,兩個鏈表仍須保持原有的結(jié)構(gòu)。
可假定整個鏈表結(jié)構(gòu)中沒有循環(huán)。
程序盡量滿足 O(n) 時間復(fù)雜度,且僅用 O(1) 內(nèi)存。

解答

這道題與【題目141. 環(huán)形鏈表】屬于同一題型,環(huán)形鏈表可以使用快慢指針判斷,這里我們同樣使用雙指針進(jìn)行判別,不過步長都是一步,讓兩個指針分別從兩個鏈表頭結(jié)點(diǎn)開始向后移動,當(dāng)其中一個指針走到鏈表末尾后,換到另一個鏈表的頭結(jié)點(diǎn)上,另一個指針也是如此,這樣如果兩個鏈表相交,則一定可以相遇,且根據(jù)數(shù)量關(guān)系可知,首次相遇的結(jié)點(diǎn)即為相交結(jié)點(diǎn)。

class Solution(object):
    def getIntersectionNode(self, headA, headB):
        """
        :type head1, head1: ListNode
        :rtype: ListNode
        """

        p1, p2 = headA, headB                       # 初始化兩個運(yùn)動結(jié)點(diǎn)p1和p2
        while p1 != p2:                             # 只要兩個結(jié)點(diǎn)還未相遇
            p1 = headB if p1 is None else p1.next   # 如果p1走到了鏈表A的末尾,則換到鏈表B上
            p2 = headA if p2 is None else p2.next   # 如果p2走到了鏈表B的末尾,則換到鏈表A上

        return p1                                   # 當(dāng)p1和p2都換到對方的鏈表上,再次相遇后第一個結(jié)點(diǎn)即為首個公共結(jié)點(diǎn),否則為None

如有疑問或建議,歡迎評論區(qū)留言~

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容