LintCode:2つのチェーンテーブルの交差
LintCode:2つのチェーンテーブルの交差
最も考えやすいのは、もちろんバブルアルゴリズムに似ています.コード:
でも意外に...タイムアウトしました.
もう1つの態様では、長いチェーンテーブルを見つけてから、|len(A)−len(B)|の距離を後方に移動することで、2つのチェーンテーブルが同じ長さを有するようになり、バブル比較を行う必要がなくなり、複雑度もnに低下する.
最も考えやすいのは、もちろんバブルアルゴリズムに似ています.コード:
# 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):
# Write your code here
if headA == None or headB == None:
return None
p1 = headA
p2 = headB
while p1.next != None:
while p2.next != None:
if p1 == p2:
return p1
p2 = p2.next
p2 = headB
p1 = p1.next
return None
でも意外に...タイムアウトしました.
もう1つの態様では、長いチェーンテーブルを見つけてから、|len(A)−len(B)|の距離を後方に移動することで、2つのチェーンテーブルが同じ長さを有するようになり、バブル比較を行う必要がなくなり、複雑度もnに低下する.
# 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):
# Write your code here
if headA == None or headB == None:
return None
p1 = headA
p2 = headB
len_a = 0
len_b = 0
while p1.next != None:
len_a += 1
p1 = p1.next
while p2.next != None:
len_b += 1
p2 = p2.next
len_c = abs(len_a - len_b)
p1 = headA
p2 = headB
if len_a >= len_b:
for i in range(len_c):
p1 = p1.next
else:
for i in range(len_c):
p2 = p2.next
while p1 != None and p2 != None:
if p1 == p2:
return p1
p1 = p1.next
p2 = p2.next
return None