kM9tJQ - Online C++ Compiler & Debugging Tool

#include <bits/stdc++.h>
using namespace std;

int main() {
	int jumlah;
	cin >> jumlah;
	vector<bool> prima(777777, true);
	vector<int> real;

	vector<int> data(jumlah);
	for (int a = 0; a < jumlah; a++) {
		cin >> data[a];
	}

	prima[0] = prima[1] = false;

	for (long long a = 2; a < 777777; a++) {
		if (prima[a]) {
			real.push_back(a);
			for (long long b = a * a; b < 777777; b += a) {
				prima[b] = false;
			}
		}
	}

	for (int a = 0; a < data.size(); a++) {
		if (data[a] >= 1 && data[a] <= real.size()) {
			cout << real[data[a] - 1] << endl;
		}
	}
}