[백준 2153] 소수 단어

문제 출처: https://www.acmicpc.net/problem/2153

에라토스테네스의 체를 이용한 문제로, 개념을 잘 모른다면 https://jow1025.tistory.com/12을 참고하자.

문제

소수란 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.를 출력한다.

풀이

에라토스테네스의 체만 구현할 줄 알면되기때문에 나머지는 주석으로 설명한다.

코드

#include<iostream>
#include<string>
//한 숫자의 최대값은 52,길이는20자까지므로50*20+1넘는 크기로 배열 선언
int arr[1100];
using namespace std;
int main()
{
	string s;
	cin >> s;

	//에라토스테네스의 체 구현
	arr[1] = 0;
	for (int i = 2; i < 1100; i++)
	{
		for (int j = i * i; j < 1100; j += i)
		{
			if (arr[j] == 1)continue;
			else arr[j] = 1;
		}
	}
	//입력한 단어를 숫자로환산할 때 소수판별
	int sum = 0;
	for (int i = 0; i < s.size(); i++)
	{
		if (s[i] >= 'a' && s[i] <= 'z')
			sum += s[i] - 'a' + 1;
		else if (s[i] >= 'A' && s[i] <= 'Z')
			sum += s[i] - 'A' + 27;
	}
	if (arr[sum])
		cout << "It is not a prime word.";
	else
		cout << "It is a prime word.";
	

}

결과