CodeForces 474B E(Contest #1)

题意:
　　给你一个数n,代表n段区间,接下来有n个数(a₁,a₂,...a_n)代表每段区间的长度,第一段区间为[1,a₁],第二段区间为[a₁+1,a₁+a₂],...第i段区间为[a_i-1+1,a_i-1+a_i]
　　接着输入一个整数m,之后有m个数,问你这m个数,每个数所在的区间编号.

分析:
　　可以采用计数排序,建立一个数组,将第i段区间[ai-1+1,ai-1+ai]赋值为i,这样输入一个数很快就可以输出它所在的区间编号;
　　因为区间中的值是有序的,所以还可以采用二分查找法.

代码如下：

 //方法一：

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <fstream>
 #include <ctime>
 #include <cmath>
 #include <cstdlib>
 #include <algorithm>
 #include <set>
 #include <map>
 #include <list>
 #include <stack>
 #include <queue>
 #include <iterator>
 #include <vector>

 using namespace std;

 #define LL long long
 #define INF 0x3f3f3f3f
 #define MON 1000000007
 #define MAXN 10000010
 #define MAXM 1000010

 const int maxn = ;
 int a[maxn];

 int main()
 {
     int n, m;
     while(scanf("%d", &n)==)
     {
         memset(a, , sizeof(a));
         int i, j;
         int cnt = ;
         int sum = ;
         int pos = ;
         for(i = ; i <= n; i++ )
         {
             pos++;
             int x;
             scanf("%d", &x);
             sum += x;   //sum的最大值为1e6超过了n的最大值,所以数组a的长度应该为1e6,不然会RE
             for(j = cnt; j <= sum; j++ )
                 a[j] = pos;
             cnt = sum+;
         }
         scanf("%d", &m);
         for(i = ; i < m; i++ )
         {
             int y;
             scanf("%d", &y);
             printf("%d\n", a[y]);
         }
     }

     return ;
 }

 

 //方法二：

 16:52:262016-07-25
 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <fstream>
 #include <ctime>
 #include <cmath>
 #include <cstdlib>
 #include <algorithm>
 #include <set>
 #include <map>
 #include <list>
 #include <stack>
 #include <queue>
 #include <iterator>
 #include <vector>

 using namespace std;

 #define LL long long
 #define INF 0x3f3f3f3f
 #define MON 1000000007
 #define MAXN 10000010
 #define MAXM 1000010

 const int maxn = ;
 int a[maxn], b[maxn];

 int lower_bound(int *A, int x, int y, int v)
 {
     int mid;
     while(x < y)
     {
         mid = (y-x)/ + x;
         if(v <= A[mid])
             y = mid;
         else
             x = mid+;
     }

     return x;
 }

 int main()
 {
     int n, m;
     while(scanf("%d", &n)==)
     {
         memset(a, , sizeof(a));
         int i;
         int sum = ;
         int c = ;
         for(i = ; i < n; i++ )
         {
             int x;
             scanf("%d", &x);
             sum += x;
             a[c++] = sum;
         }
         scanf("%d", &m);
         for(i = ; i < m; i++ )
         {
             int y;
             scanf("%d", &y);
             int cnt = lower_bound(a, , c, y);
             printf("%d\n", cnt);
         }
     }

     return ;
 }