纪中暑假培训： Date 3 ：音乐节拍

2018.07.08【2018提高组】模拟C组 :音乐节拍

题目：

FJ准备教他的奶牛弹奏一首歌曲，歌曲由N(1<=N<=50,000)种音节组成，编号为1到N，而且一定按照从1到N的顺序进行弹奏，第i种音节持续B_i(1<=B_i<=10,000)个节拍，节拍从0开始计数，因此从节拍0到节拍B_1-1弹奏的是第1种音节，从B_1到B_1+B_2-1弹奏的是第2种音节，依此类推。
　　最近奶牛对弹琴不感兴趣了，他们感觉太枯燥了。所以为了保持奶牛们注意力集中，FJ提出Q(1<=Q<=50,000)个问题，问题的格式都是“第T次节拍弹奏的是哪种音节”
　　每个问题对应一个T_i(0<=T_i<=节拍总数-1)请你帮奶牛来解决。

输入输出：

Input
　　第一行输入两个空格隔开的整数N和Q
　　第2至N+1行每行包含一个整数 B_i
　　第N+2-N+Q_1行每行包含一个整数T_i

Output
　　输出有Q行，每行输出对应问题的答案。

Sample Input
3 5
2
1
3
2
3
4
0
1

Sample Output
2
3
3
1
1

思路：二分+模拟

数据n=50000显然不能直接模拟

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;

int a[50005],ans[50005];

int main()
{
    freopen("mnotes.in","r",stdin);
    freopen("mnotes.out","w",stdout);
    int n,m,x,k;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        a[i]=a[i-1]+x;
    }
    for (int i=1;i<=m;i++)
    {
        scanf("%d",&k);
        k++;
        int l=1,r=n,mid=(l+r)/2;
        while (l<r)
        {
            if (k>a[mid]) l=mid+1;
            if (k<a[mid]) r=mid; 
            if (k==a[mid]) break;
            mid=(l+r)/2;
        }
        ans[i]=mid;
    }
    for (int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
}