UVA 11991 Easy Problem from Rujia Liu?(STL应用)

UVA 11991 Easy Problem from Rujia Liu?(STL应用)

题意:

        给你n个数(n<=10w,但是数值<=100w),现在要你回答m个查询,对于每个查询都是k和v,要求你回答原始数据中第k个v值出现的原始下标,如果不存在输出0.

分析:

        刘汝佳:训练指南P187例题.

        首先看到本题的询问时是v和k，我想有没有什么数据结构直接用X[v][k]就能解决每个询问的？即X[v]中保存的是v这个值出现过的所有下标。由于X[v]的大小不一定且需要能随机存取(取X[v]中的第k个元素为X[v][k])，所以X[v]应该是一个vector<int>。那么X是什么呢？如果X仅仅是一个数组结构，那么我们需要一个100W大小的X数组(因为v<=100W)，其实v的值并不会覆盖的那么广，所以这里我们用map<int, vector<int> > 来表示X。

        所以这里我们可以定义map<int,vector<int> > data.则每个v值对应一个v出现的所有下标构成的向量。对于每个v和k，我们可以直接通过map[v][k-1]直接输出结果(想想是不是).

AC代码(新)：

#include<map>
#include<vector>
#include<cstdio>
using namespace std;

int main()
{
    int n, m;
    while(scanf("%d%d",&n,&m)==2)
    {
        //mp是值 与 下标组成的vector 的映射
        map<int,vector<int> > mp;

        for(int i=1;i<=n;i++)
        {
            int v;
            scanf("%d",&v);
            if(mp.count(v)==0) mp[v]=vector<int>();
            mp[v].push_back(i);
        }

        while(m--)
        {
            int k, v;
            scanf("%d%d",&k,&v);
            if(mp[v].size()<k) printf("0\n");
            else printf("%d\n",mp[v][k-1]);
        }
    }
    return 0;
}

AC代码:

#include<iostream>
#include<cstdio>
#include<vector>
#include<map>
using namespace std;
map<int, vector<int> > data;
int n,m;
int main()
{
    while(scanf("%d%d",&n,&m)==2)
    {
        data.clear();
        for(int i=1;i<=n;i++)
        {
            int v;
            scanf("%d",&v);
            data[v].push_back(i);
        }
        while(m--)
        {
            int k,v;
            scanf("%d%d",&k,&v);
            if(!data.count(v) || data[v].size()<k) printf("0\n");
            else printf("%d\n",data[v][k-1]);
        }
    }
    return 0;
}