-
Notifications
You must be signed in to change notification settings - Fork 178
/
16_FindFirstCommonNode.py
62 lines (51 loc) · 1.99 KB
/
16_FindFirstCommonNode.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# 两个链表的公共节点
# 输入一个复杂链表(每个节点中有节点值,以及两个指针,一个指向下一个节点,另一个特殊指针random指向一个随机节点),
# 请对此链表进行深拷贝,并返回拷贝后的头结点。(注意,输出结果中请不要返回参数中的节点引用,否则判题程序会直接返回空)
class Solution:
# 第一个参数给比较短的链表,第二个参数给长链表的值
# def findEqual(self,):
def FindFirstCommonNode(self, pHead1, pHead2):
# 假设输入的两个链表,是同一个链表
if pHead1 == pHead2:
return pHead1
pTmp1 = pHead1
pTmp2 = pHead2
# 我们通过循环,让其中一个节点走到最后
while pTmp1 and pTmp2:
pTmp1 = pTmp1.next
pTmp2 = pTmp2.next
# 判断哪个链表先走到最后
# 假设pTmp1,还没有走完,说明pTmp2是更短的
if pTmp1:
k = 0
# 寻找链表长度之间的差值
while pTmp1:
pTmp1 = pTmp1.next
k += 1
# 我们让pTmp1先跳N步
pTmp2 = pHead2
pTmp1 = pHead1
for i in range(k):
pTmp1 = pTmp1.next
# 当找到节点相等的时候,也就是说明该节点是公共节点
while pTmp1 != pTmp2:
pTmp1 = pTmp1.next
pTmp2 = pTmp2.next
return pTmp1
# 假设pTmp2,还没有走完,说明pTmp1是更短的
if pTmp2:
k = 0
while pTmp2:
pTmp2 = pTmp2.next
k += 1
# 我们让pTmp2先跳N步
pTmp2 = pHead2
pTmp1 = pHead1
for i in range(k):
pTmp2 = pTmp2.next
while pTmp1 != pTmp2:
pTmp1 = pTmp1.next
pTmp2 = pTmp2.next
return pTmp1
if __name__ == '__main__':
print()