hdu 5233- Gunner II(BestCoder Round #42)离散化

Gunner II

Accepts: 188

Submissions: 1072

Time Limit: 2000/1000 MS (Java/Others)

Memory Limit: 65536/65536 K (Java/Others)

问题描述

很久很久以前，有一个叫Jack的枪手。他非常喜欢打猎。一天，他去了一个小树林。那儿有n只鸟，还有n棵树。第i只鸟站在第i棵树的顶端。这些树从左到右排成一条直线。每一棵树都有它的高度。Jack站在最左边那棵树的左边。当Jack在高度为H的地方向右发射一棵子弹时，站在高度为H的树上且离Jack最近的鸟儿就会落下来。
Jack会射击多次，他想知道每次射击哪只鸟儿会落下来。

输入描述

多组测试数据（大概5组），每一组的第一行给出n,m，n表示有n棵树和n只鸟，m表示Jack会射击m次。
在第二行，有n个整数h[1],h[2],h[3],…,h[n]表示这些树的高度。
在第三行，有m个整数，q[1],q[2],q[3],…,q[m]表示Jack射击的高度。

[参数约定]
所有输入均为整数。
1<=n,m<=100000(10^5)
1<=h[i],q[i]<=1000000000(10^9)

输出描述

对于每一个q[i]，在一行中输出Jack射落的鸟的id。如果没有射中鸟儿，请输出-1。
(id 从1开始编号。)

输入样例

5 5
1 2 3 4 1
1 3 1 4 2

输出样例

1
3
5
4
2

Hint

大数据输入，推荐使用快速读读入。

思路：先对数据进行离散化，把数字转化到 0 到 n−1 的区间内。然后对每一种数字串起来按照ID从小到大。可以用链表，或者Vector。查询的时候要注意如果没有把查询一起离散化的话要先判断一下数字是否可以找得到。找到之后按照下标去对应的数字串中找出第一个即可。然后删除。如果数字串中没有数字了就应该输出 −1 。空间复杂度是 O(n) ,时间复杂度是 nlogn+Qlogn

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
const int maxn=100010;
int H[maxn];
int N,M;
map<int,int> mp;
vector<int> V[maxn];
int num[maxn];
int main()
{
    while(scanf("%d%d",&N,&M)!=EOF)
    {
        mp.clear();
        for(int i=0;i<=N;i++)V[i].clear();
        int cnt=0;
        for(int i=1;i<=N;i++)
        {
            scanf("%d",&H[i]);
            if(!mp.count(H[i]))mp[H[i]]=cnt++;
            V[mp[H[i]]].push_back(i);
        }
        memset(num,0,sizeof(num));
        while(M--)
        {
            int x;
            scanf("%d",&x);
            if(!mp.count(x))printf("-1\n");
            else
            {
                int pos=mp[x];
                if(num[pos]>=V[pos].size())printf("-1\n");
                else printf("%d\n",V[pos][num[pos]]),num[pos]++;
            }
        }
    }
    return 0;
}