[백준] 4134번 다음 소수 C++ 코드

코드

// 40억까지 소수를 만들면 시간과 메모리 둘 다 터짐
// 따라서 x**0.5 범위까지 탐색하며 소수 판정하는 함수 만들며 판별
// 0,1과 같은 엣지케이스 신경쓰기
#include <iostream>
using namespace std;
typedef long long ll;
ll n,m;
bool isprime(ll x) {
  for(ll i = 2; i*i <=x; i++) {
    if(x%i == 0) return false;
  }
  return true;
}
int main() {
  cin >> n;
  while(n--) {
    cin >> m;
    if(m == 1||m==0) {
      cout << 2 << '\n';
      continue;
    }
    while(1) {
      if(isprime(m)) {
        cout << m << '\n';
        break;
      }
      m++;
    }
  }
  return 0;
}

소수판정에서는 범위를 살펴보고 0,1이 포함되어있다면 이 수들이 소수가 아니라는 체크를 해줘야한다. 다른 문제를 풀 때도 엣지 케이스를 잘 생각해봐야겠다.