网易2019笔试编程题 丰收

题目描述

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

说明

 

这道题目用前缀和加上二分搜索优化，时间复杂度可以到clogN

#include <iostream>
const int N = 100100;
using namespace std;

int arr[N];

// 维护一个数组
// 数组是一个前缀和，表示前i堆的苹果总数
// 通过二分查找可以快速找到苹果。
// 有k次问询，时间复杂度是klogn

int find(int arr[], int n, int target) {
	int left = 0, right = n - 1;
	while (left <= right) {
		int mid = left + (right - left) / 2;
		if (arr[mid] < target) {
			left = mid + 1;
		}
		else if (arr[mid] >= target && arr[mid - 1] < target)
			return mid;
		else
			right = mid - 1;
	}
	return -1;
}

int main()
{
	int n, sum = 0, temp = 0;
	int m, target;
	scanf("%d", &n);
	for (int i = 0; i < n; i++) {
		scanf("%d", &temp);
		sum += temp;
		arr[i] = sum;
	}
	scanf("%d", &m);
	while (m--) {
		scanf("%d", &target);
		printf("%d\n", find(arr, n, target)+1);
	}

}