새벽코딩

[백준] [18310] 안테나 (JAVA) 본문

알고리즘

[백준] [18310] 안테나 (JAVA)

J 코딩 2023. 2. 18. 21:27
반응형

https://www.acmicpc.net/problem/18310

18310번: 안테나

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

www.acmicpc.net


문제

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

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

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

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

입력

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

출력

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

예제 입력 1 복사

4
5 1 7 9

예제 출력 1 복사

5

※ JAVA 코드(안테나)

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

public class Main {
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		
		ArrayList<Integer> list = new ArrayList<>();
		
		int N = Integer.parseInt(br.readLine());

		st = new StringTokenizer(br.readLine());
		
		for(int i = 0; i < N; i++) {
			// 각 집의 위치
			int hPoint = Integer.parseInt(st.nextToken());
			
			list.add(hPoint);
		}
		Collections.sort(list);
		
		int idx = list.size()%2 != 0 ? list.size()/2 : list.size()/2 - 1;
		
		// 중간값 
		System.out.println(list.get(idx));
	}
}

※ 생각정리 (안테나)
이 문제를 처음 봤을 때 논리적으로 한위치에 여러개의 집이 존재할 수 있다는 문장을 보고 문제를 너무 어렵게 생각했었다.
하지만 한 위치에 여러개의 집이 존재한다 한들 정렬된 집들의 중간값이 안테나가 존재할 수 있는 위치가 된다.
예를 들어 설명하자면,
 
1, 5, 7, 9, 9, 9, 9, 9 라는 수가 입력되었을때 안테나가 존재할수 있는 위치는 1, 5, 7,9, 9, 9, 9, 9 해당 숫자인 9번이 된다.
예시에서는 1, 5, 7, 9의 집 위치가 주어졌고 1,5, 7, 9 집들 중에서 5가 중간 값이 된다.
쉬운 문제를 너무 오래 고민하고 있었다. 비록 문제는 간단한 코드로 해결 할 수 있었지만 다양한 방법으로 그리디 문제를 풀어보아 생각하는 힘을 기를 수 있었다.
하지만, 때로는 문제를 너무 어렵게가 아닌 간단하게 생각해볼 필요가 있다. ㅎㅎ
 
-새벽코딩-

반응형
Comments