[编程题]丰收

[编程题]丰收

又到了丰收的季节，恰逢小易去牛牛的果园里游玩。
牛牛常说他对整个果园的每个地方都了如指掌，小易不太相信，所以他想考考 牛牛。
在果园里有N堆苹果，每堆苹果的数量为a _i，小易希望知道从左往右数第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:

1
5
3

思路

1. 输入按照递增保存，例如输入：2,1,3。保存vector中的值为：2,3,6。便于后期处理。
2. 使用lower_bound查到第一个大于$q_i$的下标，最后输出即可。
3. lower_bound使用链接：C++ STL中的Binary search（二分查找）

代码

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n, temp, m;
    cin>> n;
    vector<int> v(n);
    for (int i = 0; i < n; ++i) {
        cin>> temp;
        if (i == 0)
            v[i] = temp;
        else
            v[i] = temp + v[i-1];
    }
    cin>> m;
    for (int i = 0; i < m; i++) {
        cin>> temp;
        int pos = lower_bound(v.begin(), v.end(), temp) - v.begin();
        cout<< pos+1<< endl;
    }
    return 0;
}