题目描述
又到了丰收的季节,恰逢小易去牛牛的果园里游玩。
牛牛常说他对整个果园的每个地方都了如指掌,小易不太相信,所以他想考考牛牛。
在果园里有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
思路:
首先记录一下前缀和,得到递增的和数组sum,下面就是看输入的q在sum的哪个元素里。再次遍历sum会导致O(m*n)超时,
注意到sum有序,因此采用二分查找。由于这次要寻找一个区间(a<sum=<b,则答案为b)而不是具体值,所以如果while条件是left<right的话可能会变成死循环,我们让right-left=1时退出循环,此时确定区间为[a,b],确定结果就很简单了,当num<=a时,在第left+1堆,否则在第right+1堆。(num>a之前已经有num<b了,上次递归的条件)
#include <iostream>
#include <algorithm>
#include <limits.h>
using namespace std;
int binarySearch(int* s,int length,int t)
{
int left=0,right=length-1,mid;
while(left+1<right)
{
mid=(left+right)/2;
if(s[mid]==t)
return mid+1;
else if(s[mid]>t)
right=mid;
else
left=mid;
}
return (t<=s[left])?left+1:right+1;
}
int main()
{
int n;
int m;
cin>>n;
int* a=new int [n];
int* sum=new int [n];
for(int i=0;i<n;++i)
cin>>a[i];
sum[0]=a[0];
for(int i=1;i<n;++i)
sum[i]=sum[i-1]+a[i];
cin>>m;
while(m--)
{
int q;
cin>>q;
cout<<binarySearch(sum,n,q)<<endl;
}
return 0;
}