网易笔试题：丰收（python解答，bisect）

题目描述

又到了丰收的季节，恰逢小易去牛牛的果园里游玩。
牛牛常说他对整个果园的每个地方都了如指掌，小易不太相信，所以他想考考牛牛。
在果园里有N堆苹果，每堆苹果的数量为ai，小易希望知道从左往右数第x个苹果是属于哪一堆的。
牛牛觉得这个问题太简单，所以希望你来替他回答。

输入描述:

第一行一个数n(1 <= n <= 105)。
第二行n个数ai(1 <= ai <= 1000)，表示从左往右数第i堆有多少苹果
第三行一个数m(1 <= m <= 105)，表示有m次询问。
第四行m个数qi，表示小易希望知道第qi个苹果属于哪一堆。

输出描述:

m行，第i行输出第qi个苹果属于哪一堆。

示例1

输入

5
2 7 3 4 9
3
1 25 11

输出

1
5
3

python答案

from bisect import bisect_left
n = int(input())
a = list(map(lambda x:int(x),list(input().split(' '))))
m = int(input())
q = list(map(lambda x:int(x),list(input().split(' '))))
for i in range(1,len(a)):
    a[i] += a[i-1]
for i in q:
    print(bisect_left(a,i)+1)

这题思路不难，只不过不用二分查找的话会超时，Python 有一个 bisect 模块，用于维护有序列表。bisect 模块实现了二分法用于插入元素到有序列表。那么bisect_left方法会返回要被插入元素的索引，若该元素已存在那么返回该元素左边的索引。