PS/BOJ
[알고리즘] 백준 1990번: 소수인팰린드롬 (C++), 이런 문제는 어떡하지
도비(Doby)
2021. 11. 21. 06:53
https://www.acmicpc.net/problem/1990
1990번: 소수인팰린드롬
151은 소수이면서 동시에 팰린드롬이기 때문에 소수인 팰린드롬이다. 팰린드롬이란 앞으로 읽어나 뒤로 읽으나 같은 수를 말한다. 예를 들어 1234는 앞으로 읽으면 1234지만, 뒤로 읽으면 4321이 되
www.acmicpc.net
솔루션
1) 에라토스테네스의 체를 통해 입력받은 b보다 작거나 같은 소수들을 구한다.
2) a와 b 사이의 범위를 만족시키기 위해 이분 탐색을 통해 시작과 끝 범위를 찾는다.
3) bool 타입의 팰린드롬 판별 함수를 통해 소수 중에 팰린드롬을 찾는다.
4) -1 출력, 종료
>> 다음 솔루션은 메모리 초과 발생
>> 게시판 질문 글에서 10,000,000을 넘어가는 팰린드롬인 소수는 없다고 한다.
(https://www.acmicpc.net/board/view/1961)
그래서 소수를 구할 때 b가 10,000,000을 넘는 경우 10,000,000을 넘어가지 않게 조건을 걸어주었다.
if (b >= 10000000) {
b = 10000000;
}
그런데 저런 경우로 메모리 초과가 되는 경우는 어떻게 해야 할까?
10,000,000을 넘어가는 소수인 팰린드롬이 없다는 것을 알아내는 것도 능력 중 일부인가?
혹은 팰린드롬 판별 함수를 더 작게 줄일 수 있는 걸까
[AC 코드]
#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>
#include <stack>
#include <string>
using namespace std;
vector<int> num;
vector<int> prime;
int a, b;
void primeSave() {
if (b >= 10000000) {
b = 10000000;
}
for (int i = 0; i <= b; i++) {
num.push_back(i);
}
for (int i = 2; i <= sqrt(b); i++) {
if (num[i] != 0) {
for (int j = i + i; j <= b; j += i) {
num[j] = 0;
}
}
}
for (int i = 2; i < num.size(); i++) {
if (num[i] != 0) {
prime.push_back(num[i]);
}
}
}
bool isRight(string temp) {
if (temp.size() == 1) {
return 1;
}
stack<char> s;
if (temp.size() % 2 == 0) {
for (int i = 0; i < (temp.size() / 2); i++) {
s.push(temp[i]);
}
for (int i = temp.size() / 2; i < temp.size(); i++) {
if (s.top() == temp[i]) {
s.pop();
}
else {
return 0;
}
}
return 1;
}
else {
for (int i = 0; i < (temp.size() / 2); i++) {
s.push(temp[i]);
}
for (int i = (temp.size() / 2) + 1; i < temp.size(); i++) {
if (s.top() == temp[i]) {
s.pop();
}
else {
return 0;
}
}
return 1;
}
}
int main() {
cin >> a >> b;
primeSave();
int start = lower_bound(prime.begin(), prime.end(), a) - prime.begin();
int end = upper_bound(prime.begin(), prime.end(), b) - prime.begin();
for (int i = start; i < end; i++) {
if (isRight(to_string(prime[i]))) {
cout << prime[i] << '\n';
}
}
cout << -1;
return 0;
}