2019网易笔试题C++--丰收

题目描述

又到了丰收的季节，恰好小易去牛牛的果园里游玩。 牛牛常说他多整个果园的每个地方都了如指掌，小易不太相信，所以他想考考牛牛。 在果园里有N堆苹果，每堆苹果的数量为ai,小易希望知道从左往右数第x个苹果是属于哪一堆的。 牛牛觉得问题太简单了，所以希望你来替他回答。

输入描述:

第一行一个数n(1<=n<=10^5) 第二行n个数ai(1<=ai<=1000),表示从左往右数第i堆有多少苹果 第三行一个数m(1<=m<=10^5),表示有m次询问 第四行m个数qi,表示小易希望知道第qi个苹果属于哪一堆。

输出描述:

m行，第i行输出第qi个苹果属于哪一堆。

输入例子1:

5
2 7 3 4 9
3
1 25 11

输出例子1:

1
5
3


该题的解题思路：求累加和，在累加和中查找，如果采用暴力查找只能通过30%的测试用例，时间复杂度太高，查找的时候可以使用二分查找！
代码：

//2018年8月11日15:55:16-丰收-采用二分查找就可以了-但是要注意边界条件

#include <iostream>
using namespace std;

int main() {
    int n;
    cin >> n;
    int a[n];
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
    }
    int m;
    cin >> m;
    int q[m];
    for (int i = 0; i < m; ++i) {
        cin >> q[i];
    }

    int sum[n];
    int res[m];
    sum[0] = a[0];
    for (int i = 1; i < n; ++i) {
        sum[i] = sum[i - 1] + a[i];
    }
    for (int i = 0; i < m; ++i) {
        int j = 0, k = n - 1;
        while (j < k) {
            int mid = (j + k) / 2;
            if (q[i] > sum[mid]) {
                j = mid + 1;
            } else{
                k = mid;
            }
        }
        res[i] = k + 1;
    }

    //输出
    for(int i = 0; i < m; ++i) {
        cout << res[i] << endl;
    }
    return 0;
}