억억단을 외우자

#include <string> #include <string.h> #include <vector> using namespace std; int A[5000001]; int B[5000001]; vector<int> solution(int e, vector<int> starts) { memset(A, 0, sizeof(int)); memset(B, 0, sizeof(int)); for (int i = 1; i <= e; i++) { int n = i; while (n <= e) { A[n]++; n += i; } } // B[i]는 i부터 e까지 중 A[i]가 가장 큰 i값을 저장합니다. B[e] = e; for (int i = e-1; i > 0; --i) { if (A[i] >= A[B[i+1]]) { B[i] = i; } else { B[i] = B[i+1]; } } vector<int> answer; for (int s: starts) { answer.push_back(B[s]); } return answer; }