用C++写洛谷P3156 【深基15.例1】询问学号

题目描述

有 n(n ≤ 2×10^6) 名同学陆陆续续进入教室。我们知道每名同学的学号（在 1 到 10^9 之间），按进教室的顺序给出。上课了，老师想知道第 ii 个进入教室的同学的学号是什么（最先进入教室的同学 i=1），询问次数不超过 10^5 次。

输入格式

第一行 2 个整数 n 和 m，表示学生个数和询问次数。

第二行 n 个整数，表示按顺序进入教室的学号。

第三行 m 个整数，表示询问第几个进入教室的同学。

输出格式

m 个整数表示答案，用换行隔开。

输入输出样例

输入 #1复制

10 3
1 9 2 60 8 17 11 4 5 14
1 5 9

输出 #1复制

1
8
5

代码如下：

#include <iostream>
using namespace std;
int main(){
	int n,m,a;
	cin >> n >> m;
	int nums[n+1];
	for (int i = 1;i <= n;i++){
		cin >> nums[i];
	}
	for (int i = 0;i < m;i++){
		cin >> a;
		cout << nums[a] << endl;
	}
	return 0;
}