题目描述:
又到了丰收的季节,恰逢小易去牛牛的果园里游玩。
牛牛常说他对整个果园的每个地方都了如指掌,小易不太相信,所以他想考考牛牛。
在果园里有N堆苹果,每堆苹果的数量为ai,小易希望知道从左往右数第x个苹果是属于哪一堆的。
牛牛觉得这个问题太简单,所以希望你来替他回答。
输入描述:
第一行一个数n(1 <= n <= 105)。
第二行n个数ai(1 <= ai <= 1000),表示从左往右数第i堆有多少苹果
第三行一个数m(1 <= m <= 105),表示有m次询问。
第四行m个数qi,表示小易希望知道第qi个苹果属于哪一堆。
输出描述:
m行,第i行输出第qi个苹果属于哪一堆。
分析:非常简单直观的数学题. 令sum[i]表示前i堆苹果的总数,如果 sum[i-1] < q[i] <= sum[i],则表示第 q[i] 个苹果属于第 i 堆,返回 i.
关键点:二分查找
#include <iostream>
using namespace std;
int n, a[100002], m, q[100002]; // 如题
int sum[100002];
int get_index(int x){
int left = 1, right =n;
while( left < right-1 ){
if( sum[left] > x ) return left;
if( sum[right] < x ) return right+1;
if( sum[left] == x ) return left;
if( sum[right] == x ) return right;
if( sum[(left+right)/2] == x )
return (left+right)/2;
else if( sum[(left+right)/2] > x )
right = (left+right)/2 ;
else
left = (left+right)/2 ;
}
return right;
}
int main()
{
input
scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&a[i]); // 下标从1开始
scanf("%d",&m); for(int i=1;i<=m;i++) scanf("%d",&q[i]); // 下标从1开始
for(int i=1;i<=n;i++){
sum[i] = sum[i-1] + a[i];
}
for(int i=1;i<=m;i++){
printf("%d\n",get_index(q[i]));
}
return 0;
}