[이것이 코딩테스트다] 안테나

🗃️문제 설명

일직선 상의 마을에 여러 채의 집이 위치해 있다. 이중에서 특정 위치의 집에 특별히 한 개의 안테나를 설치하기로 결정했다. 효율성을 위해 안테나로부터 모든 집까지의 거리의 총 합이 최소가 되도록 설치하려고 한다. 이 때 안테나는 집이 위치한 곳에만 설치할 수 있고, 논리적으로 동일한 위치에 여러 개의 집이 존재하는 것이 가능하다.

집들의 위치 값이 주어질 때, 안테나를 설치할 위치를 선택하는 프로그램을 작성하시오.

예를 들어 N=4이고, 각 위치가 1, 5, 7, 9일 때를 가정하자.

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

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

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

🖥️코드

import sys
input = sys.stdin.readline

N = int(input())
data = list(map(int, input().split()))
data.sort()

if N % 2 == 0:
    f1 = data[(N // 2) - 1]
    f2 = data[N // 2]
    print(min(f1, f2))
else:
    print(data[N // 2])

🧠아이디어

알고리즘 유형 : 정렬

전체 집의 수가 홀수일 때와 짝수일 때를 비교해야 한다.

또한, 안테나를 설치할 수 있는 위치 값이 최소 2개 이상 나올 경우 가장 작은 값을 출력하도록 해야 한다.

🔒문제 출처 및 참고 코드

이것이 코딩테스트다 with 파이썬 - 안테나
모범 답안 - 안테나