【map】【HDOJ】5233 Gunner II

http://acm.hdu.edu.cn/showproblem.php?pid=5233

 

外面有很多树，每棵树顶上有一只鸟，一个数组按从近到远的顺序列出这些树的高度（也就是鸟的高度）

猎人开始从不同高度打枪，子弹不能穿过鸟，也就是在同一高度上只有最近的鸟会被打下来。

给出一个数组表示猎人每次打枪的高度，要求出每次打落的鸟所在的树的编号（从1开始编号），如果这个高度上没有鸟输出-1

 

放一个MAP表示每个高度上从近到远的鸟的位置编号，

猎人每次射击就将该高度的定位数向后移，直到定位数大于鸟的数量，就输出-1

 

#include<bits/stdc++.h>
using namespace std;

typedef vector<int> vec;
map<int, vec> b;
map<int, int> pos;
int n, m, shot;

int main(){
    while(~scanf("%d%d", &n, &m)){
        b.clear();
        pos.clear();
        for(int i = 1; i <= n; i++){
            int h;
            scanf("%d", &h);
            b[h].push_back(i);
            pos[h] = 0;
        }
        map<int, vec>::iterator it;
        while(m--){
            scanf("%d", &shot);
            it = b.find(shot);
            if(it != b.end()){
                int p = it->first;
                int l = it->second.size();
                if(pos[p] < l){
                    printf("%d\n", it->second[pos[p]]);
                    pos[p]++;
                }
                else printf("-1\n");
            }
            else{
                printf("-1\n");
            }
        }
    }

    return 0;
}