面试了某蚂蚁,笔试题目,没有编译器,直接白板写代码,够溜的,用了暴力破解,有点问题,没有达到最优
题目:给你一个数组和一个数字num,找出数组中两个数a 和b,使得a+b 与 num 的差的绝对值最小。比如输入[8,3,6,1] 和13,输出8 和6。
输入的代码:
array1,num1=input().split(" ") array=[] array=array1[1:-1].split(",") array=list(map(int,array)) num=int(num1) print(array) print(num)
解决问题的部分,主要思路是:先把array排序,用两个指针,一个在左边,一个在右边,两个指针所指的数字之和,如果大于13,那么j-- 如果小于13,那么i++
这样的话,就可以比暴力破解更快
array.sort() length=len(array) i=0 j=length-1 deta=0 temp_min=num while i<j-1: temp_sum=array[i]+array[j] deta=abs(temp_sum-num) if deta<temp_min: temp_min=deta index_i=i index_j=j i+=1 if deta>temp_min: j-=1 print(array[i],array[j])