（未AC）洛谷P3156 【深基15.例1】询问学号

题目描述

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

输入格式

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

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

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

输出格式

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

输入输出样例

输入

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

输出

1
8
5

---

第一版代码：（显示所有数据超时）

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

Q：感觉跟线性表没很大关系，原思路是先将输入的学号存在一个数组中，只需要读取该数组所在的位置即可，但为什么会数据超时？

题解代码： 

#include<bits/stdc++.h> 
using namespace std;
int num[2000005];
int main()
{
	int n, m, a;
	while(cin >> n >> m)
	{
		memset(num, 0, sizeof(num));
		for(int i = 1; i <= n; i++)	//2e6
			cin >> num[i];
		for(int i = 1; i <= m; i++)
		{
			cin >> a;
			cout << num[a] << endl;
		}
	}
    return 0;
}

 

---

第二版代码：（建立队列）