来自Rosalind上的一个简单的题目,但是却足足困扰我三天(哭哭脸)
最后用了一个比较讨巧的方法解决了需求。
个人觉得代码是有问题的,以后发现问题了在修正吧~
# coding=utf-8
import os
os.chdir('C:/workspace')
fh = open('rosalind_bins.txt')
read_fh = fh.readlines()
len_array = int(read_fh[0][:-1])
len_list = int(read_fh[1][:-1])
A_array = [int(x) for x in read_fh[2].split()]
A_array.append('None')
k_list = [int(x) for x in read_fh[3].split()]
fh.close()
def bin_search(array,k):
low = 0
height = len(array)-1
while low < height:
mid = (low+height)/2
if array[mid] < k:
low = mid + 1
elif array[mid] > k:
height = mid
else:
return mid+1
return -1
for k in k_list:
print bin_search(A_array,k),