牛客网 丰收

题目描述

又到了丰收的季节，恰逢小易去牛牛的果园里游玩。
牛牛常说他对整个果园的每个地方都了如指掌，小易不太相信，所以他想考考牛牛。
在果园里有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

解题思路:

放在第几堆，意味着寻找qi属于某一个苹果堆区间[i,i+ai)即 i<=qi<i+ai,涉及到了大小比较
设计苹果堆数列[1,1+a1,1+a1+a2,…]每个元素代表每个苹果堆的开始序号
因此可以用寻找插入位置的方法来找到第qi个苹果属于的苹果堆

python的话，可以用bisect模块，而java需自行编写二分法

python:

n = int(input())
a = list(map(int,input().split(' ')))
m = int(input())
q = list(map(int,input().split(' ')))

t = [1]
for i in range(n):
    t.append(t[i]+a[i])
import bisect

for i in q:
    idx = bisect.bisect(t,i)
    print(idx)

java的二分法

	public static int _bisect(int[] a,int target) {
		int size = a.length;
		int start =0;
		int end =size-1;
		while (start<=end) {
			int mid = (start+end)/2;
			if(start==end) {
				return start+1;
			}
			else if(a[mid]==target) {
				return mid;
			}
			else if(a[mid]<target) {
				start = mid+1;
			}
			else if (a[mid]>target) {
				end = mid-1;
			}
		}
		return start+1;
	}