[Kotlin] 백준 1978 소수 찾기

Posted by Space_Jin
2025. 3. 8. 12:31 Programming/Algorithm
728x90
반응형

문제

주어진 수 N개 중에서 소수가 몇 개인지 찾아서 출력하는 프로그램을 작성하시오.

입력

첫 줄에 수의 개수 N이 주어진다. N은 100이하이다. 다음으로 N개의 수가 주어지는데 수는 1,000 이하의 자연수이다.

출력

주어진 수들 중 소수의 개수를 출력한다.

예제 입력 1 복사

4
1 3 5 7

예제 출력 1 복사

3

 

풀이

소수를 찾는 효과적인 방법은 "에라토스테네스의 체"

 

에라토스테네스의 체 핵심은 "소수의 배수는 소수가 아니다" 이다.

 

1부터 최댓값(여기서는 1000)까지 미리 모든 소수를 구해놓는다.

1~N의 제곱근까지 두번의 반복문으로 최대 N까지 수행되며, O(N)의 시간복잡도를 가진다.

 

코틀린 풀이 코드
import java.util.*

fun main() = with(System.`in`.bufferedReader()) {
	// 에라토스테네스의 체
    val max = 1000
    val N = readLine().toInt()	// 입력으로 주어지는 N 값
    val list = readLine().split(" ").map{ it.toInt() }	// 입력으로 주어지는 리스트
    val primeArray = BooleanArray(max + 1) { true }	// 0~1000까지 숫자가 소수인지 여부를 저장
    var result = 0	// 주어진 리스트 중 소수의 갯수
    
    // 초기값 세팅, 0과 1은 소수가 아님 예외처리
    primeArray[0] = false
    primeArray[1] = false
    
    // 2부터 최댓값(1000)의 루트 값까지 순환
    for(i in 2..Math.sqrt(max.toDouble()).toInt()) {
        // 현재 값이 소수라면 현재 값의 배수는 모두 소수가 아님
        if(primeArray[i]){
        	// 본인의 제곱부터 최댓값(1000)까지의 배수 처리
            // 본인보다 작은 값을 곱하는 케이스는 선행되었을 것이므로 수행하지 않음 (4x3의 경우, 3x4로 수행되었음)
            for(j in i * i..max step i) {
            	primeArray[ j ] = false                
            }
        }
    }
   
   	// 입력된 리스트를 순회하며 소수의 갯수를 확인 후 출력
    list.forEach{ if(primeArray[it]) result++ }
    println(result) 
}
728x90
반응형

[Kotlin] 백준 1292 - 쉽게 푸는 문제

Posted by Space_Jin
2024. 11. 13. 22:47 Programming/Algorithm
728x90
반응형

1. 문제

동호는 내년에 초등학교를 입학한다. 그래서 동호 어머니는 수학 선행 학습을 위해 쉽게 푸는 문제를 동호에게 주었다.

이 문제는 다음과 같다. 1을 한 번, 2를 두 번, 3을 세 번, 이런 식으로 1 2 2 3 3 3 4 4 4 4 5 .. 이러한 수열을 만들고 어느 일정한 구간을 주면 그 구간의 합을 구하는 것이다.

하지만 동호는 현재 더 어려운 문제를 푸느라 바쁘기에 우리가 동호를 도와주자.

2. 입력

첫째 줄에 구간의 시작과 끝을 나타내는 정수 A, B(1 ≤ A ≤ B ≤ 1,000)가 주어진다. 즉, 수열에서 A번째 숫자부터 B번째 숫자까지 합을 구하면 된다.

3. 풀이 (생각의 흐름)

1. 구간의 합 -> 3~7 구간의 합이란, 7까지 누적된 합 - 2까지 누적된 합과 동일하다

2. 3~7구간의 합 = 7 누적 합 - 2 누적 합

3. 누적 합이란 0부터 N까지 순차적으로 순회하면서 적용할 수 있다. -> Big O(N), N의 최대 값은 1,000이므로 수용한다.

4. 제출 소스

import java.util.*

fun main() = with(System.`in`.bufferedReader()) {
    val (s, e) = readLine().split(" ").map{ it.toInt() }	// 입력된 두 값을 받기 
    var n = 1	// 현재 구간의 값
    var c = 1	// 현재 구간의 값이 나온 횟수
    
    val arr = IntArray(e + 1)	// 구간의 누적합을 담을 배열
    arr[0] = 0					// 1구간을 index 1로 사용하므로 0구간의 합은 0

	// 구간 1부터 e까지 순회
    for (i : Int in 1..e){
        arr[i] = arr[i-1] + n	// 현재 구간의 누적합 = 이전 구간의 누적합 + 현재 구간의 값
        c++						// 현재 구간의 값이 나온 횟수를 추가
        if(c > n){				// 현재 구간의 값이 나온 횟수가 현재 구간의 값을 초과하다면, 값을 올리고 갯수를 1로 초기화
            n++
            c = 1
        }
    }
    
    println(arr[e] - arr[s-1])	// 출력: 끝 누적합 - 시작 지점 앞의 누적합
}

 

728x90
반응형