[BOJ B2][C++] 백준 2153번: 소수 단어

https://www.acmicpc.net/problem/2153

2153번: 소수 단어

 

문제

소수란 1과 자기 자신으로만 나누어떨어지는 수를 말한다. 예를 들면 1, 2, 3, 5, 17, 101, 10007 등이 소수이다. 이 문제에서는 편의상 1도 소수로 하자.

알파벳 대소문자로 이루어진 영어 단어가 하나 있을 때, a를 1로, b를 2로, …, z를 26으로, A를 27로, …, Z를 52로 하여 그 합을 구한다. 예를 들어 cyworld는 합을 구하면 100이 되고, abcd는 10이 된다.

이와 같이 구한 수가 소수인 경우, 그 단어를 소수 단어라고 한다. 단어가 주어졌을 때, 그 단어가 소수 단어인지 판별하는 프로그램을 작성하시오.

입력

첫째 줄에 단어가 주어진다. 단어의 길이는 20자 이하이다. 주어지는 단어는 알파벳 소문자와 대문자만으로 이루어져 있다.

출력

아래의 예제와 같은 형식으로 출력을 한다. 소수 단어인 경우에는 It is a prime word.를, 아닌 경우에는 It is not a prime word.를 출력한다.

 

풀이

에라토스테네스의 체를 구현한다면 소수는 쉽게 판별 가능하다

하지만!

메모리 초과 가 뜬다면..

함수를 따로 만들지 말고 메인함수에서 해결하자....

이거때문에 오래걸렸다 ㅠㅠ

 

// Authored by : seondal
// Co-authored by : -

// #include <bits/stdc++.h>
#include <iostream>
#include <vector>

using namespace std;

int main() {
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    
    // v[i] : 숫자 i가 소수인가?
    vector<bool> isPrime(1040, true);
    for(int i=2; i<=1040; i++) {
        if(!isPrime[i])
            continue;
        for(int j=i*i; j<=1040; j+=i) {
            if(!isPrime[j])
                continue;
            isPrime[j] = false;
        }
    }
    
    // 입력
    string word;
    cin >> word;

    // 단어의 합 구하기
    int cnt = 0;
    for(int i=0; i<word.size(); i++){
        if(word[i] < 90)  // 대문자
            cnt += word[i]-'A'+27;
        else  // 소문자
            cnt += word[i]-'a'+1;
    }
    
    // 출력
    if(isPrime[cnt])
        cout << "It is a prime word.";
    else
        cout << "It is not a prime word.";
    
    return 0;
}

/*
 */