EOJ2890. 询问队员身高

单点时限: 2.0 sec

内存限制: 256 MB

ECNU 篮球队教练陈老师选拔了一批篮球队员。对于篮球运动，队员身高是重要因素之一，陈老师的记忆力不好，他经常会向助理教练询问身高第 i 高的人是谁。助理教练不胜其烦，想请你帮忙写个程序回答陈老师的询问。

备注：每个队员的身高都各不相同。

输入格式

第一行是一个整数 t，表示测试数据组数。

对于每组测试数据：

第一行是两个整数 n,m (1≤n≤50,1≤m≤50)，n 表示队员的人数，m 表示询问的次数

第二行有 n 个整数，a1,a2,a3,…,an，依次为第 1 个到第 n 个队员的身高 ，(1≤i≤n，170≤ai≤230)。

第三行有 m 个整数，b1,b2,b3,…,bm，bj 表示教练陈老师的第 j 个询问，询问身高第 bj 高的队员编号 (1≤j≤m,1≤bj≤n)

输出格式

每组测试数据输出一行，每行输出 m 个整数，依次表示所求队员的编号。两个整数之间用一个空格分开，每行的最后有一个换行符。

样例

input

2
3 1
228 205 190
3
5 3
223 179 217 195 193
3 4 5

output

3
4 5 2

#include <iostream>
#include <bits/stdc++.h>
#include <math.h>
using namespace std;
class SingleJob {
	public:
		int personNum, queryNum;
		
		vector<pair<int, int>> Height;
		vector<int > Query;
};

int cmp(pair<int, int> a, pair<int, int> b){
	return (a.first > b.first);
}

int main() {
	int jobs, temp;
	SingleJob J[10];

	// 输入
	cin >> jobs;
	int seq = 0;	//中间变量，用来每输入一个身高，随之记录其序号
	for(int i = 0; i < jobs; i++) {
		cin >> J[i].personNum >> J[i].queryNum;
		for(int j=0; j<J[i].personNum; j++){
			cin >> temp;
			seq++;
			J[i].Height.insert(J[i].Height.end(), {temp, seq});
		}
		seq = 0;
		for(int j=0; j<J[i].queryNum; j++){
			cin >> temp;
			J[i].Query.insert(J[i].Query.end(), temp);
		}		
	}

	// 输出
	for(int i = 0; i < jobs; i++) {
//		cout << "case #" << i << ":" << endl;
		sort(J[i].Height.begin(), J[i].Height.end(), cmp);	// 降序排序，所用规则是cmp()
		for(int j=0; j<J[i].queryNum; j++){
			temp = J[i].Query[j];
			cout << J[i].Height[temp-1].second << ((j==J[i].queryNum-1)?"":" ");
		}
		cout << endl;
	}
	// 以下是 对 J[i] 全体队员的 <身高，序号> 的打印
//	for(int i = 0; i < jobs; i++){
//		for(int j=0; j<J[i].personNum; j++){
//			cout << "<"  << J[i].Height[j].first << " , "<< J[i].Height[j].second << ">  " ;
//		}
//		cout << endl;
//	}
	
}