백준 16401 과자 나눠주기 Kotlin

: ) YOUNG·2023년 5월 26일
1

알고리즘

목록 보기
204/371
post-thumbnail

백준 16401번
https://www.acmicpc.net/problem/16401

문제




생각하기


  • 이분 탐색 문제이다.
    • 과자의 길이를 이분 탐색을 통해서 찾아내면 된다.
  • 모든 조카에게 같은 길이의 막대과자를 나눠줄 수 없다면, 0을 출력해야 한다는 예외를 처리해야 한다.

동작




코드



Kotlin


import java.io.*
import java.util.*

private var N = 0
private var M = 0
private lateinit var arr: IntArray

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.`out`))
    val sb = StringBuilder()

    var st = StringTokenizer(br.readLine())
    M = st.nextToken().toInt() // 조카의 수
    N = st.nextToken().toInt() // 과자의 수
    arr = IntArray(N)

    var max = Int.MIN_VALUE
    st = StringTokenizer(br.readLine())
    for (i in 0 until N) {
        arr[i] = st.nextToken().toInt()
        max = Math.max(max, arr[i])
    }

    var result = binarySearch(1, max)
    if(result == -1) {
        result = 0
    }

    sb.append(result)
    bw.write(sb.toString())
    bw.close()
} // End of main

private fun binarySearch(low: Int, high: Int): Int {
    if (low > high) {
        return -1
    }

    val mid = (low + high) / 2
    val cnt = countCheck(mid)
    if (cnt >= M) {
        // 과자를 나눠서 나온 갯수가 조카의 수 보다 많을 때
    	// 갯수를 줄여야 하기 때문에 mid값이 더 높아져야 함
        val temp = binarySearch(mid + 1, high)
        if (temp == -1) {
            return mid
        } else {
            return temp
        }
    } else {
        return binarySearch(low, mid - 1)
    }
} // End of binarySearch

private fun countCheck(mid: Int): Int {
    var count = 0
    arr.forEach {
        count += it / mid
    }
    return count
} // End of check

0개의 댓글