hdu 5233 Gunner II 离散化

题目链接:

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

题意:

有n只鸟，还有n棵树。第i只鸟站在第i棵树的顶端。这些树从左到右排成一条直线。每一棵树都有它的高度。Jack站在最左边那棵树的左边。当Jack在高度为H的地方向右发射一棵子弹时，站在高度为H的树上且离Jack最近的鸟儿就会落下来。 Jack会射击多次，他想知道每次射击哪只鸟儿会落下来。

题解:

离散化

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define MS(a) memset(a,0,sizeof(a))
#define MP make_pair
#define PB push_back
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 2e5+10;

struct node{
    int x,id;
}h[maxn];

vector<int> k,dp[maxn];
map<int,int> H;
int q[maxn],cnt[maxn];

void init(){
    MS(h); MS(q); MS(cnt);
    H.clear(); k.clear();
    for(int i=0; i<maxn; i++)
        dp[i].clear();
}

int main(){
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
        init();
        for(int i=0; i<n; i++){
            h[i].x = read();
            h[i].id = i+1;
            k.push_back(h[i].x);
        }
        for(int i=0; i<m; i++){
            q[i] = read();
            k.push_back(q[i]);
        }
        sort(k.begin(),k.end());
        k.erase(unique(k.begin(),k.end()),k.end());
        for(int i=0; i<(int)k.size(); i++){
            H[k[i]] = i;
        }
        for(int i=0; i<n; i++)
            dp[H[h[i].x]].push_back(h[i].id);
        for(int i=0; i<m; i++){
            if(cnt[H[q[i]]] >= (int)dp[H[q[i]]].size())
                cout << -1 << endl;
            else{
                cout << dp[H[q[i]]][cnt[H[q[i]]]] << endl;
                cnt[H[q[i]]]++;
            }
        }

    }

    return 0;
}