洛谷——P1918 保龄球

DL 算缘分算得很烦闷，所以常常到体育馆去打保龄球解闷。因为他保龄球已经打了几十年了，所以技术上不成问题，于是他就想玩点新花招。

DL 的视力真的很不错，竟然能够数清楚在他前方十米左右每个位置的瓶子的数量。他突然发现这是一个炫耀自己好视力的借口——他看清远方瓶子的个数后从某个位置发球，这样就能打倒一定数量的瓶子。

1 OOO

2 OOOO

3 O

4 OO

如上图，每个“O”代表一个瓶子。如果 DL 想要打倒 3 个瓶子就在 1 位置发球，想要打倒 4 个瓶子就在 2 位置发球。

现在他想要打倒 m 个瓶子。他告诉你每个位置的瓶子数，请你给他一个发球位置。

输入格式

输入文件名为 bowling.in。

第一行包含一个正整数 n，表示位置数。

第二行包含 n 个正整数，第 i 个数。表示第 i 个位置的瓶子数，保证各个位置的瓶子数不同。

第三行包含一个正整数 Q，表示 DL 发球的次数。

第四行至文件末尾，每行包含一个正整数 m，表示 DL 需要打倒 m 个瓶子。

输出格式

输出文件名为 bowling.out。

共 Q 行。每行包含一个整数，第 i 行的整数表示 DL 第 i 次的发球位置。若无解，则输出 0。

输入输出样例

输入 #1复制

5
1 2 4 3 5
2
4
7

输出 #1复制

3
0

说明/提示

【数据范围】

对于 50%的数据，1 ≤ n,Q ≤ 1000，1 ≤ai,M ≤ 10^5

对于 100%的数据，1 ≤ n,Q ≤ 100000，1 ≤ai,M ≤ 10^9

解题方法：结构体 + 二分。

Code 1:

#include <iostream>
#include <stdio.h>
#include <algorithm>

using namespace std;

const int N = 1000010;

int n, m, x;

struct stu
{
    int a, b;
}s[N];

bool cmp(stu p, stu q)
{
    return p.a < q.a;
}

signed main()
{
    scanf("%d", &n);
    for(int i = 0; i < n; i ++ )
    {
        scanf("%d", &s[i].a);
        s[i].b = i + 1;
    }
    sort(s, s + n, cmp);
    scanf("%d", &m);
    for(int i = 1; i <= m; i ++ )
    {
        scanf("%d", &x);
        int l = 0, r = n - 1;
        while(l < r)
        {
            int mid = l + r >> 1;
            if(s[mid].a >= x)  r = mid;
            else  l = mid + 1;
        }
        if(s[l].a == x) cout << s[l].b << endl;
        else  cout << 0 << endl;
    }
}

方法二：STL. 

Code 2:

#include <iostream>
#include <stdio.h>
#include <map>

using namespace std;

int n, m, x;
map<int, int> mp;

signed main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i ++ ) 
    {
        scanf("%d", &x);
        mp[x] = i;
    }
    scanf("%d", &m);
    for(int i = 1; i <= m; i ++ )
    {
        scanf("%d", &x);
        cout << mp[x] << endl;
    }
}