标签:
题目描述:请写一个程序,找到两个单链表最开始的交叉节点。注意事项如果两个链表没有交叉,返回null;在返回结果后,两个链表仍须保持原有的结构;可假定整个链表结构中没有循环。
两个链表交叉,其实就是在某一时刻,两个链表指向了同一个节点。那么,如果两个链表等长,这道题就毫无难度了,我们直接分别遍历两个链表,看在哪个节点处,两个链表的指向的节点是一样的。
但是这道题并没有说链表等长,那也好办,我把他们变成等长的。具体思路可以这样描述:
1. 统计两个链表的长度,得到长度差
2. 设置两个指针,分别指向两个链表的头结点
3. 令较长的链表的指针先走与长度差相等的步数,这样一来,如果从现在走到的这个节点开始,同时遍历两个链表,他们第一次指向同一节点时,就是最开始交叉的位置。
那样例来说,令b链表先走一步,到达b2处,从b2开始,同时遍历两个链表,知道两个链表都指向c1时,停止循环,返回c1.
所以,直接写出代码就行:
# Definition for singly-linked list. # class ListNode: # def __init__(self, x): # self.val = x # self.next = None class Solution: # @param headA: the first list # @param headB: the second list # @return: a ListNode def getIntersectionNode(self, headA, headB): curA, curB = headA, headB countA, countB = 0, 0 # 两个循环,统计两个链表的长度 while curA: countA += 1 curA = curA.next while curB: countB += 1 curB = curB.next # 计算长度差 gap = abs(countA - countB) # 令长的链表先走gap长 if countA >= countB: while gap != 0: headA = headA.next gap -= 1 else: while gap != 0: headB = headB.next gap -= 1 # 开始遍历,看是否能指向同一节点 curA, curB = headA, headB while curA: if curA == curB: return curA curA = curA.next curB = curB.next return None # Write your code here
标签:
原文地址:http://blog.csdn.net/guoziqing506/article/details/51355492