Computer/Algorithm

Daily Algorithm - 소수

kentakang 2018. 4. 21. 15:28
반응형

문제

문제1) 소수 (중등1)

자연수 M과 N이 주어질 때 M이상 N이하의 자연수 중 소수인 것을 모두 골라 이들 소수의 합과 최소값을 찾는 프로그램을 작성하시오.

예를 들어 M=60, N=100인 경우 60이상 100이하의 자연수 중 소수는 61, 67, 71, 73, 79, 83, 89, 97 총 8개가 있으므로, 이들 소수의 합은 620이고, 최소값은 61이 된다.

입력

첫 째 줄에 M이, 둘째 줄에 N이 주어진다. M과 N은 10000이하의 자연수이며 M은 N보다 같거나 작다.

출력

M이상 N이하의 자연수 중 소수인 것을 모두 찾아 첫째 줄에 그 합을, 둘째 줄에 그 중 최소값을 출력한다.

단, M이상 N이하의 자연수 중 소수가 없을 경우는 첫째 줄에 -1을 출력한다.

예제 입력

60
100

예제 출력

620
61

풀이

#include <stdio.h>
#include <stdlib.h>

void eratos(int n, int m)
{
int *arr = malloc(sizeof(int) * m);
int sum = 0;
int min = m;
int cnt = 0;

for (int i = n; i <= m; i++)
{
arr[i] = i;
}

for (int i = 2; i <= m; i++)
{
for (int j = n; j <= m; j++)
{
if ((arr[j] != i && arr[j] % i == 0) || arr[j] == 1)
arr[j] = 0;
}
}

for (int i = n; i <= m; i++)
{
if (arr[i] != 0)
{
sum += arr[i];
if (arr[i] < min)
min = arr[i];

cnt++;
}
}

if (cnt == 0)
printf("-1");
else
printf("%d\n%d", sum, min);
}

int main()
{
int n, m;

scanf("%d %d", &n, &m);
eratos(n, m);
}


에라토스테네스의 채를 이용해서 빠르게 소수를 구했다.

나는 n과 m 사이에 소수가 없을 때 -1을 출력하는 부분을 추가해줬는데, -1을 출력해주지 않아도 정답 처리된다.

테스트 케이스 중 소수가 없는 케이스가 없는 것으로 판단된다.


문제 출처 : http://codeup.kr/JudgeOnline/problem.php?id=4566



반응형