网易2019笔试-丰收

Posted 2023-02-26 一米阳光213

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