정렬 : 안테나[백준 18310, Kotlin]

[문제]

일직선 상의 마을에 여러 채의 집이 위치해 있다. 이중에서 특정 위치의 집에 특별히 한 개의 안테나를 설치하기로 결정했다. 효율성을 위해 안테나로부터 모든 집까지의 거리의 총 합이 최소가 되도록 설치하려고 한다. 이 때 안테나는 집이 위치한 곳에만 설치할 수 있고, 논리적으로 동일한 위치에 여러 개의 집이 존재하는 것이 가능하다.
집들의 위치 값이 주어질 때, 안테나를 설치할 위치를 선택하는 프로그램을 작성하시오.
예를 들어 N=4이고, 각 위치가 1, 5, 7, 9일 때를 가정하자.

이 경우 5의 위치에 설치했을 때, 안테나로부터 모든 집까지의 거리의 총 합이 (4+0+2+4)=10으로, 최소가 된다.

 

[입력 조건]

• 첫째 줄에 집의 수 N이 자연수로 주어진다. (1≤N≤200,000) 둘째 줄에 N채의 집에 위치가 공백을 기준으로 구분되어 1이상 100,000이하의 자연수로 주어진다.

 

[출력 조건]

• 첫째 줄에 안테나를 설치할 위치의 값을 출력한다. 단, 안테나를 설치할 수 있는 위치 값으로 여러 개의 값이 도출될 경우 가장 작은 값을 출력한다.

 

해결한 방법

import java.io.BufferedReader
import java.io.InputStreamReader
import kotlin.math.abs

fun main(){
    val br = BufferedReader(InputStreamReader(System.`in`))
    val n = br.readLine().toInt()
    val numList = br.readLine().split(" ").map{ it.toInt() }.sorted()
    // 인덱스 설정
    if(n % 2 == 0){ // 짝수 일 때 0123
        val leftValue = getDistance(n / 2 - 1, numList)
        val rightValue = getDistance(n / 2, numList)
        if(rightValue < leftValue){
            print(numList[n / 2])
        }else{
            print(numList[n / 2 - 1])
        }
    }else{ // 홀수 일 때
        print(numList[n / 2])
    }
}
fun getDistance(index : Int, numList: List<Int>): Int{
    var result = 0
    repeat(numList.size){ j ->
        result += abs(numList[index] - numList[j]) // 거리를 모음
    }
    return result
}

최선의 경우를 생각하였을 때 입력 받은 주소의 중간값이 가장 효율적으로 측정될 것 같았다. 홀수 일 경우 딱 절반이 나눠지지만, 짝수 일 경우 절반이 나눠지지 않아, 중간값과 중간에서 하나 뺀 값을 비교하여 값을 도출했다. 

 

다른 사람이 해결한 방법 

import java.io.BufferedReader
import java.io.InputStreamReader
import java.util.StringTokenizer

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val n = br.readLine().toInt()
    val st = StringTokenizer(br.readLine())
    val locOfHouses = IntArray(n) { st.nextToken().toInt() }

    locOfHouses.sort()
    val locOfAntenna = locOfHouses[locOfHouses.size / 2 - 1 + n % 2]

    println(locOfAntenna)
}

다른사람이 풀이한 방법을 보니, 짝수일 경우 절반이 나눠지지 않지만, 절반에서 하나 뺀 값이 최솟값이 된다.