数组的第k个v的下标+map的使用+UVA-11991

Easy Problem from Rujia Liu?
用空间换时间的最极端的做法就是用data[k][v]时间复杂度O(1),空间复杂度O(k*v),无法承受。

map<int,vector<int>> kv;map的查询,赋值耗时都是O(log),vector<>是O(1),时间复杂度(logn),总时间复杂度为O((n+m)*logn)

#include<stdio.h>
#include<iostream>
#include<cmath>
#include<math.h>
#include<string>
#include<string.h>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
#define ms0(a) memset(a,0,sizeof(a))
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
map<int,vector<int>> kv;
int n,m,k,v;
int main(){
    while (~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)
        {
            int x;
            scanf("%d",&x);
            if(!kv.count(x)) //如果还没有的话,先创建键值对
                kv[x]=vector<int>();
            kv[x].push_back(i);
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&k,&v);
            if(!kv.count(v) || kv[v].size()<k)
                printf("%d\n",0);
            else
                printf("%d\n",kv[v][k-1]);
        }
    }
    return 0;
}
发布了67 篇原创文章 · 获赞 0 · 访问量 1494

猜你喜欢

转载自blog.csdn.net/qq_44846324/article/details/104610015