11991 - Easy Problem from Rujia Liu?(抽象数据结构)

时间:2022-02-04 10:29:01

该题m很大,要求我们可以快速的求出第k个v的下标。   我们最先想到的肯定是用数组下标直接访问d[k][v] , 但是n和m都太大了,数组开不下。    但是幸运的是我们可以用map和vector来组织一个新的数据结构, 以实现这个目的。

需要注意的是,map用数组下标访问的复杂度实际上的log(n),这是map查询的复杂度,vector的随机存取是O(1)。所以总的时间复杂度是O(n*log(n));

细节参见代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 100000 + 5;
const int INF = 1000000000;
int n,m,x,y;
map<int, vector<int> > a;
int main() {
    while(~scanf("%d%d",&n,&m)) {
        a.clear();
        for(int i = 1; i <= n; i++) {
            scanf("%d",&x);
            if(!a.count(x)) a[x] = vector<int> ();
            a[x].push_back(i);
        }
        while(m--) {
            scanf("%d%d",&x,&y);
            if(!a.count(y) || a[y].size() < x) printf("0\n");
            else printf("%d\n",a[y][x-1]) ;
        }
    }
    return 0;
}