Kotlin 🍬 백준 14단계 :: 10815 번

숫자 카드

문제   |

  숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 가지고 있는지 아닌지를 구하는 프로그램을 작성하시오.

 

입력   |

  첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 두 숫자 카드에 같은 수가 적혀있는 경우는 없다.

  셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 가지고 있는 숫자 카드인지 아닌지를 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다.

 

출력   | 

  첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 가지고 있으면 1을, 아니면 0을 공백으로 구분해 출력한다.

 

 

풀이  |

  M개의 숫자를 담은 array로 for문을 돌려 N개의 array 내부에 해당 값이 있는지 비교한다.

 

 

답안  |

fun main() {
    val N = readLine()!!.toInt()
    val nArr = readLine()!!.split(" ").map(String::toInt)
    val M = readLine()!!.toInt()
    val mArr = readLine()!!.split(" ").map(String::toInt)
    
    val result = Array(M) {0}
    mArr.forEachIndexed { i, n ->
        if (n in nArr) {
            result[i] = 1
        }
    }
    print(result.joinToString(" "))
}

 

^ 시간초과가 발생한 코드

 

array를 Set으로 변환하여 빠르게 요소를 확인할 수 있도록 수정한 코드

import java.io.BufferedReader
import java.io.InputStreamReader

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))

    val N = br.readLine().toInt()
    val nArr = br.readLine().split(" ").map(String::toInt).toSet() // 변환 후 Set으로 저장
    val M = br.readLine().toInt()
    val mArr = br.readLine().split(" ").map(String::toInt)

    val result = IntArray(M)
    for (i in mArr.indices) {
        if (mArr[i] in nArr) {
            result[i] = 1
        } else {
            result[i] = 0
        }
    }

    print(result.joinToString(" "))
}