문제
일직선 상의 마을에 여러 채의 집이 위치해 있다. 이중에서 특정 위치의 집에 특별히 한 개의 안테나를 설치하기로 결정했다. 효율성을 위해 안테나로부터 모든 집까지의 거리의 총 합이 최소가 되도록 설치하려고 한다. 이 때 안테나는 집이 위치한 곳에만 설치할 수 있고, 논리적으로 동일한 위치에 여러 개의 집이 존재하는 것이 가능하다.
집들의 위치 값이 주어질 때, 안테나를 설치할 위치를 선택하는 프로그램을 작성하시오.
예를 들어 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().rstrip())
houses = sorted([*map(int, input().rstrip().split(" "))])
print(houses[(n-1)//2])
'개발 > 알고리즘' 카테고리의 다른 글
[BOJ] 11650 좌표 정렬하기 (0) | 2020.11.11 |
---|---|
[BOJ] 10989 수 정렬하기 2 (0) | 2020.11.10 |
[BOJ] 18405 경쟁적 전염 (0) | 2020.11.07 |
[BOJ] 1699 제곱수의 합 (0) | 2020.11.06 |
[BOJ] 1010 다리 놓기 (0) | 2020.11.03 |