1070: PIPI的数组

最新推荐文章于 2025-12-03 11:47:41 发布

原创最新推荐文章于 2025-12-03 11:47:41 发布 · 588 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#动态规划 #c语言 #stl

本文介绍了一个数组查询问题，通过使用两个map容器实现对数组中特定元素第k次出现位置的高效查询。具体而言，一个map用于记录每个数字出现的次数，另一个map则用于存储指定数字第k次出现的具体位置。

1070: PIPI的数组

题目描述

PIPI现在有一个全部为正整数组成的数组，他想知道元素 v 第 k次出现在数组中的位置，胖虎能帮帮PIPI吗?
PIPI是个好奇心很重的人，他会有m次询问哦~

输入

输入包含多组测试样例。
对于每一组测试样例，第一行包含两个正整数 n,m .代表数组的长度和询问的次数。 (1<=n,m<=105)
第二行包括n个正整数,每个正整数 v < 106。
接下来包含m组询问，每一行包含两个数字 k 和 v ,代表元素v第k次出现的下标。

输出

对于每一组测试用例，输出m次询问的结果。如果没有，输出0.

样例输入

8 4
1 3 2 2 4 3 2 1
1 3
2 4
3 2
4 2

样例输出

思路

使用两个 map 容器，map<int（v）, int> cnt 用来存储当前数字 v 出现的次数，map<pair<int（v), int （k）>, int> mp 用来存储数字 v 第 k 次出现的位置。

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n, m, k, v;
	int num[100010];
	while(scanf("%d%d", &n, &m) != EOF)
	{
		map<int, int> cnt;
		map<pair<int, int>, int> mp;
		for(int i = 0; i < n; i++)
			scanf("%d", &num[i]);
		for(int i = 0; i < n; i++)	
		{
			cnt[num[i]]++;
			mp[{num[i], cnt[num[i]]}] = i + 1;
		}
		for(int i = 0; i < m; i++)
		{
			scanf("%d%d", &k, &v);
			printf("%d\n", mp[{v, k}]);
		}	
	}
	return 0;
}