题目:
输入一个已经按升序排序过的数组和一个数字,
在数组中查找两个数,使得它们的和正好是输入的那个数字。
要求时间复杂度是O(n)。如果有多对数字的和等于输入的数字,输出任意一对即可。例如输入数组1、2、4、7、11、15和数字15。由于4+11=15,因此输出4和11。
思路:
1 第一种思路,可以把数字存在数组里,比如数组中最大值是15,那么就开一个长度未15的数组1 存在a[1]里 15存在a[15]里;这样用15-a[1]判断里面是否有值就可以了。缺点就是浪费空间。
2 因为是求两个数,时间复杂度是O(n),还是排过顺序的数组,那么可以从头和从尾同时找;从尾开始的tail下标大于sum,则tail左移;如果tail和head相加小于sum,则tail右移;指导头尾两个数相加等于求和;或者tail大于head为止;
代码如下:
'''
题目:输入一个已经按升序排序过的数组和一个数字,
在数组中查找两个数,使得它们的和正好是输入的那个数字。
要求时间复杂度是O(n)。如果有多对数字的和等于输入的数字,输出任意一对即可。
例如输入数组1、2、4、7、11、15和数字15。由于4+11=15,因此输出4和11。
'''
if __name__ == "__main__":
input_data = [0,1,2,4,7,11,15]
sum =6
head = 0
tail = len(input_data) - 1
while True:
if head > tail:
break
if input_data[tail] >sum:
tail = tail - 1
else:
if (input_data[tail] + input_data[head]) < sum:
head = head + 1
else:
if (input_data[tail] + input_data[head]) == sum:
print input_data[head]
print input_data[tail]
break
输出 2 4
--------------------------------------------------