【网易】丰收

题目描述
又到了丰收的季节,恰逢小易去牛牛的果园里游玩。
牛牛常说他对整个果园的每个地方都了如指掌,小易不太相信,所以他想考考牛牛。
在果园里有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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值