要求:
给定以非递减顺序排序的三个数组,找出这三个数组的公共元素。如ar1 = [2,5,12,20,45,85],ar2 = [16,19,20,85,200], ar3 = [3,4,15,20,39,72,85,190],那么其公共元素为[20,85]。
分析:
假设当前遍历的三个数组元素分别为ar1[i]、ar2[j]、ar3[k],则存在以下几种可能性:
(1)如果ar1[i]、ar2[j]、ar3[k]相等,则说明当前遍历的元素是三个数组的公共元素,可以直接打印出来,然后通过执行i+,j+,k+,使得三个数组同时向后移动,此时继续遍历各数组后面的元素。
(2)如果ar1[i]<ar2[j],则执行 i+ 来遍历ar1中后面的元素,因为ar1[i] 不可能是三个数组的公共元素。
(3)如果ar2[j]<ar3[k],同理可以通过 j+ 来遍历ar2后面的元素。
(4)如果前面的条件都不满足,说明ar1[i]>ar2[j]而且ar2[j]>ar3[k],此时可以通过 k+ 来继续遍历ar3后面的元素。
实现代码:
# -*- coding:utf-8 -*-
def findCommon(ar1,ar2,ar3):
i = 0
j = 0
k = 0
n1 = len(ar1)
n2 = len(ar2)
n3 = len(ar3)
#遍历三个数组
while i < n1 and j < n2 and k < n3:
#找到了公共元素
if ar1[i] == ar2[j] and ar2[j] == ar3[k]:
print(ar1[i],'',end="")
i += 1
j += 1
k += 1
#ar1[i]不可能是公共元素
elif ar1[i] < ar2[j]:
i += 1
#ar2[j]不可能是公共元素
elif ar2[j] < ar3[k]:
j += 1
#ar3[k]不可能是公共元素
else:
k += 1
if __name__ == "__main__":
ar1 = [2, 5, 12, 20, 45, 85]
ar2 = [16, 19, 20, 85, 200]
ar3 = [3, 4, 15, 20, 39, 72, 85, 190]
findCommon(ar1,ar2,ar3)
运行结果:
20 85
性能分析:
时间复杂度为O(N1+N2+N3)。