题意:输入一个长度为n个数组(元素为[1, 1,000,000]之间的正整数),接着输入m个询问(1 <= n, m <= 100,000),第个询问为两个数k, v(1 <= k <= n,1 <= v <= 1,000,000),问数组中第k个v的下标是多少?
——>>如果每个询问都搜索一次,猜想超时过不了!于是将其弄成一一映射,用data[v][k]存第k个v的下标,询问的时候O(1)的输出,就不会超时喽!用STL中的map与vector来做就好,汝佳牛真的太神啦!谢谢!
#include <iostream>
#include <map>
#include <vector>
using namespace std;
int main()
{
int n, m, i, temp, v;
unsigned k;
while(cin>>n>>m)
{
map<int, vector<int> > data;
for(i = 1; i <= n; i++)
{
cin>>temp;
if(!data.count(temp)) //如果不存在键值temp,则返回0
data[temp] = vector<int>(); //对键值为temp的键开向量
data[temp].push_back(i); //存入下标
}
for(i = 0; i < m; i++)
{
cin>>k>>v;
if(!data.count(v) || k > data[v].size()) //如果不存在键值v,或者是第k个v已经不在向量中的话
cout<<0<<endl;
else
cout<<data[v][k-1]<<endl; //向量的下标是从0开始的,所以第k个数对应着向量中的下标k-1
}
}
return 0;
}
做第二次:
#include <cstdio>
#include <vector>
using namespace std;
const int maxn = 1000000 + 10;
vector<int> data[maxn];
int main()
{
int n, m, i, temp, v;
unsigned k;
while(scanf("%d%d", &n, &m) == 2){
for(i = 0; i < maxn; i++) data[i].clear();
for(i = 1; i <= n; i++){
scanf("%d", &temp);
data[temp].push_back(i); //存入下标
}
for(i = 0; i < m; i++){
scanf("%d%d", &k, &v);
if(data[v].empty() || k > data[v].size()) //如果不存在v,或者是第k个v已经不在向量中的话
printf("0\n");
else
printf("%d\n", data[v][k-1]); //向量的下标是从0开始的,所以第k个数对应着向量中的下标k-1
}
}
return 0;
}