Codeforces Round #271 (Div. 2)B. Worms

原题链接https://codeforces.com/problemset/problem/474/Bhttps://codeforces.com/problemset/problem/474/B题目大意：有n个按一定顺序的洞a1, a2, ..., an，有m只虫子，每只虫子都必须从按顺序从a1开始爬，输出最后停下的时候落在哪个洞上。

 

解题思路：

1、前缀和。但是因为m中的元素无序，所以对每只虫子都要从头开始判断，时间复杂度为O()。

会导致TEL。

参考代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
int a[N];
int main(){
    int n ,m ;
    
    cin >>n;

    for(int i = 1;i <= n ;i ++){
        int t;
        scanf("%d",&t);
        a[i] = a[i-1] + t;
    }
    
    cin >> m;
    for(int i = 0 ;i < m ;i ++){
        int k;
        scanf("%d",&k) ;
        for(int j = 1;j <= n ;j ++)
        if( k <= a[j] ){printf("%d\n",j);break;}
    }
    return 0;
}

2、标记虫子所有可能爬过长度（值为洞的序号）。

AC代码：

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n, a[1000010] = {0}, m, b, t;
    scanf("%d",&n);
    int xl = 1;
    for (int i = 1; i <= n; i++){
        scanf("%d",&t);
    for (int j =1 ; j <= t; j++)
        a[xl++] = i;
    }
    scanf("%d",&m);
    while (m--){
        scanf("%d",&b);
        printf("%d\n",a[b]);
     }
    return 0;
}