网易面试题 丰收

丰收

时间限制：1秒

空间限制：262144K

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

1
5
3

#include <iostream>
#include <vector>
using namespace  std;

int main(){
	int n,m;
	cin>>n;
	vector<int> arr(n+1);
	
	int  sum = 0,num;
	arr[0] = 0;
	for(int i=1;i<=n;i++){
		cin>>num;
		sum += num;
		arr[i] = sum;
	}
	
	cin>>m;
	vector<int> q(m+1);
	for(int i=1;i<=m;i++){
		cin>>q[i];
	}
	
	for(int i=1;i<=m;i++){
		
		int l = 1,r = n;//利用二分查找，否则时间复杂度太高会造成超时 
		while(l <= r){
			int mid = (l+r)/2;
			if(arr[mid-1] < q[i] && arr[mid]>= q[i]){
				cout<<mid<<endl;
				break;
			}
			else if(arr[mid-1] >= q[i] && arr[mid]>= q[i]){
				r = mid - 1;
			}	
			else
				l = mid + 1;
		}
	}
	
	
	return 0;
}