백준 18310번. 안테나

연성·2020년 9월 28일
0

코딩테스트

목록 보기
32/261

백준 18310번. 안테나

1. 문제

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

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

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

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

2. 입력

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

3. 출력

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

4. 풀이

  • 처음보자마자 중간 어디 값이 가장 작을 거란 생각이 들었다.
  • 근데 중간 어디라고만 생각이 들었지 그 중 어딘지 몰랐다.
  • 그래서 전부 값을 비교해보기로 했다.
  • 중복도 있으니까 자동 정렬도 되고 중복도 제거해주는 짱짱 set을 쓰기로 했다
  • 시간초과
  • 그런 거 없고 그냥 정렬하고 중간 인덱스 값 찾으면 되는 거였다.

5. 처음 코드와 달라진 점

  • set 썼다가 시간초과나서 지웠다

6. 코드

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;


int main() {
	
	vector<int> v;
	
	int n;
	cin >> n;

	for (int i = 0; i < n; i++){
		int house;
		cin >> house;
		v.push_back(house);
	}
	
	sort(v.begin(), v.end());

	cout << v[(n - 1) / 2] << endl;
}

0개의 댓글