保序排序 2.0

题目描述

一个保序排序，满足任意两个相等的元素，其排序后在数组中的前后相对顺序和排序之前一样。例如对于数组 [2,1,4,2] 排序得到 [1,2,2,4] ，而保序排序则满足: 排序后数组中第一个 $2$ 就是排序前数组中第一个 $2$ (位于排序前数组头部)，排序后数组中第二个 $2$ 是排序前数组中第二个 $2$ (位于排序前数组尾部)。

可以证明，给定一个数组，在保序排序后，数组中每个元素在排序前数组中的下标是确定且唯一的。

现在，给定一个正整数数组，请你对其进行保序排序（从小到大）。

之后给出m个查询，每个查询询问在原来数组（排序前）位置为x的元素（下标从 $1$ 开始），在排序后的位置（下标从 $1$ 开始）。

输入描述

第一行两个正整数 $n$ 和 $m$，表示数组的长度和查询的次数。

后面一行 $n$ 个正整数，表示数组中的每一个正整数。

之后$m$行，每行一个正整数，表示查询原来数组（排序前）位置为x的元素。

输出描述

$m$行，每行一个正整数，表示查询原来数组（排序前）位置为x的元素，在排序后的位置（下标从 $1$ 开始）。

输入样例

3 3
3 2 2
2
3
1

输出样例

1
2
3

//参考答案
#include <iostream>
#include <algorithm>
using namespace std;

struct Pair{
	int a;
	int num;
};
Pair pairs[10010];

// 稳定排序
bool cmp(Pair a1, Pair a2) {
	if (a1.a < a2.a)
		return true;
	else if (a1.a == a2.a && a1.num < a2.num)
		return true;
	return false;
}

int map[10010]; // 记录原来第几位的元素在排序后的位置
int n, m;

void sort_map() {
	sort(pairs+1, pairs+1+n, cmp);
	// 记录原来第几位的元素在排序后的位置
	for (int j = 1; j <= n; j++)
		map[pairs[j].num] = j;
}

int main() {

	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		cin >> pairs[i].a;
		pairs[i].num = i;
	}
	sort_map(); // 开始用O(nlogn)时间算法排序一次

	int x;
	for (int i = 0; i < m; i++) {
		scanf("%d", &x);
		printf("%d\n", map[x]);
	}

	return 0;
}