科大讯飞求第K个素数问题

现有q组数，针对每一组，第K个素数是多少？

输入：第一行一个q，表示数组大小；接下来q行表示第k个素数；

输出：q行表示第k个素数的值

#include <iostream>  
#include <cmath>  
#include <algorithm>

using namespace std;
int a[100000], prime[100000];

bool IsPrime(int num)   //num为我们要判断的数  
{
	for (int i = 2; i <= sqrt(num); i++)    
	{
		if (num%i == 0)
		{
			return false;
		}
	}
	return true;
}

int main()
{
	int q;
	cin >> q;
	for (int i = 0; i<q; i++)
		cin >> a[i];

	int *p = max_element(a, a + q);

	int num = 0;
	int c = 2;
	while (num < *p) {
		if (IsPrime(c)) {
			prime[++num] = c;
		}
		++c;
	}

	for (int j = 0; j<q; j++)
		cout << prime[a[j]] << endl;
	system("pause");
	return 0;
}