2019年 网易笔试测试开发岗--苹果丰收

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

输入描述:第一行一个数n(1<=n<=10^5)

                  第二行n个数ai(1<=ai<=1000),表示从左往右数第i堆有多少苹果

                 第三行一个数m(1<=m<=10^5),表示有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>
#include<algorithm>
using namespace std;

int main()
{
	int n;
	cin >> n;
	vector<int> apple_num(n);
	for (int i = 0; i < n; i++){
		cin >> apple_num[i];
	}
	int request_num;
	cin >> request_num;

	vector<int>request(request_num);
	for (int i = 0; i < request_num; i++){
		cin >> request[i];

	}
	vector<int>result;
	vector<int> total_num;
	int sum = 0;
	for (int i = 0; i < n; i++){
		sum += apple_num[i];
		total_num.push_back(sum);
		//if (sum > )
	}
	for (int i = 0; i < request_num; i++){
		int low = 0;
		int high = n - 1;
		while (low <high){
			int mid = (low + high) / 2;
			if (total_num[mid] < request[i])//比这个数字小,说明在后面堆里
				low = mid + 1;
			else         //比这个数字大,说明在这个堆以及这个堆之前
				high = mid;
		}
		result.push_back(high + 1);
	
	}
	for (auto a : result){
		cout << a << endl;
	}
	system("pause");
	return 0;
}

 参考:http://www.cnblogs.com/gousheng/p/9461953.html

猜你喜欢

转载自blog.csdn.net/liugg2016/article/details/81703490