题目:
输入 月季月季....海棠海棠...
请返回最后一个月季的索引。
思路:使用二分法查找,注意边界条件。
使用递归的方法。使用递归的方法时,一定要注意递归函数可以重复使用,或者说我们合适地分解了问题。
我们用1表示月季,0表示海棠。
python 代码
inl=[11111111111000000]
设计迭代过程:
def getIndex(inl)
mid = int(len(inl)/2)
if inl[mid]==1:
return mid + getIndex(inl[mid+1:])+1 #假设getIndex(inL2)能正确返回inL2中最后一个月季的位置,但是索引是以0开始的,所以要加上1
else:
return getIndex(inl[:mid])
看起来很easy嘛,其实是有很大的陷阱的,需要完善一些bug
inl=[] ==> a
inl=[0...] == >a
当不存在月季的时候要返回一个约定的数字,表明月季不存在。
(红色强调的是mid指向的元素)
举例 1 1 1 0 0
这时右半边[0,0经过函数getIndex时,我们希望输出0
所以a设置为-1
验证一下
举例1 1 1 1 0
假设[1,0]经过函数getIndex可以正确返回1的下标0,mid加上getIndex([1,0])需要再加上1得到整个数组的最后一个1的索引。
举例1 0 0 0 0
变成解答1 0 0的问题
又变成解答1 0的问题
所以我们要设置当inl元素数量不多的时候的特殊返回情况
def getIndex(inl):
if len(inl) == 0:
return -1
else:
if inl[0] == 0:
return -1
else:
mid = int(len(inl)/2)
#inl[:mid], inl[mid+1:]
if inl[mid] == 1:
return mid + getIndex(inl[mid+1:]) + 1
else:
return getIndex(inl[:mid])
def main():
inl1 = [1,1,1]
print getIndex(inl1)
inl1 = []
print getIndex(inl1)
inl1 = [0,0,0]
print getIndex(inl1)
inl1 = [1,1,1,0,0,0]
print getIndex(inl1)
inl1 = [1]
print getIndex(inl1)
inl1 = [0]
print getIndex(inl1)
if __name__=='__main__':
main()