HUAWEI 机试题:两个集合输出距离内数字

题目描述:

同一个数轴x有两个点的集合A={A1,A2,…,Am}和B={B1,B2,…,Bm}
A(i)和B(j)均为正整数
A、B已经按照从小到大排好序,AB均不为空
给定一个距离R 正整数,列出同时满足如下条件的(A(i),B(j))数对:
A(i)<=B(j)
A(i),B(j)之间距离小于等于R
在满足1,2的情况下每个A(i)只需输出距离最近的B(j)
输出结果按A(i)从小到大排序
输入描述:
第一行三个正整数m n R
第二行m个正整数 表示集合A
第三行n个正整数 表示集合B
输入限制 1<=R<=100000,1<=n,m<=100000,1<= A(i),B(j) <= 1000000000
输出描述:
每组数对输出一行 A(i)和B(j)
以空格隔开

输入

4 5 5
1 5 5 10
1 3 8 8 20

输出

1 1
5 8
5 8

def find_item(rank, array_a: list, array_b: list):
	"""
	寻找最近元素
	:param rank: 最大距离
	:param array_a: a数组
	:param array_b: b数组
	:return:
	"""
	a_point = 0
	b_point = 0
	len_a = len(array_a)
	len_b = len(array_b)
	ans = []
	while a_point < len_a and b_point < len_b:
		val = array_b[b_point] - array_a[a_point]
		if 0 <= val <= rank:
			ans.append((array_a[a_point], array_b[b_point]))
			a_point += 1
			b_point += 1
		elif val < 0:
			b_point += 1
		else:
			a_point += 1
	return ans

你可能感兴趣的:(算法,python)