본문 바로가기
코딩테스트

[이진 탐색] 백준 2110번 공유기 설치 자바 (Java)

by Enhydra lutris 2023. 1. 22.

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

 

1920번: 수 찾기

첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들

www.acmicpc.net

문제

도현이의 집 N개가 수직선 위에 있다. 각각의 집의 좌표는 x1, ..., xN이고, 집 여러개가 같은 좌표를 가지는 일은 없다.

도현이는 언제 어디서나 와이파이를 즐기기 위해서 집에 공유기 C개를 설치하려고 한다. 최대한 많은 곳에서 와이파이를 사용하려고 하기 때문에, 한 집에는 공유기를 하나만 설치할 수 있고, 가장 인접한 두 공유기 사이의 거리를 가능한 크게 하여 설치하려고 한다.

C개의 공유기를 N개의 집에 적당히 설치해서, 가장 인접한 두 공유기 사이의 거리를 최대로 하는 프로그램을 작성하시오.

입력

첫째 줄에 집의 개수 N (2 ≤ N ≤ 200,000)과 공유기의 개수 C (2 ≤ C ≤ N)이 하나 이상의 빈 칸을 사이에 두고 주어진다. 둘째 줄부터 N개의 줄에는 집의 좌표를 나타내는 xi (0 ≤ xi ≤ 1,000,000,000)가 한 줄에 하나씩 주어진다.

출력

첫째 줄에 가장 인접한 두 공유기 사이의 최대 거리를 출력한다.

코드

package BaekJoon.BinarySearch;

import java.util.Arrays;
import java.util.Scanner;

public class No2110 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int N = sc.nextInt();
        int C = sc.nextInt();
        int result = 0;

        int[] arr = new int[N];

        for(int i = 0; i < N; i++) {
            arr[i] = sc.nextInt();
        }

        Arrays.sort(arr); //이분 탐색을 위한 정렬

        int left = 1;
        int right = arr[N-1] - arr[0]; //최댓값은 제일 앞에 있는 집과 제일 뒤에 있는 집의 거리

        while (left <= right){
            int mid = (left + right) / 2;
            int count = 1; //첫번째 집은 무조건 설치하기 때문에 시작이 1이다.
            int startLocate = arr[0];


            for (int i = 0; i < N; ++i){
                if(arr[i] - startLocate >= mid){
                    count++; //설치 개수
                    startLocate = arr[i]; //여기서 부터 거리 다시 계산
                }
            }

            if (C > count){
                right = mid - 1; //공유기를 다 설치하지 못한경우 거리를 더 짧게 하여 다시 탐색
            }
            else {
                result = mid;
                left = mid + 1; //공유기가 다 설치된 경우 거리가 더 긴 상태에서 설치할 수 있는지 탐색
            }
        }
        System.out.println(result);
    }
}

풀이

1. for문을 이용하여 mid가 두 집사이의 최소 거리일때 설치할 수 있는 공유기의 값(count)의 최대를 구한다.

2. 설치하려 했던 공유기의 개수보다 count가 작다면 최소 거리를 더 짧게 하여 더 많은 공유기를 설치 할 수 있게 한다.

3. 설치하려 했던 공유기 만큼 다 설치 했다면, 최소 거리를 더 늘려가며 C만큼 설치 할 수 있는 최대 길이를 찾는다.

 

후기

다른 유형들은 골드가 실버보다 확실히 어려웠는데 이진 탐색은 실버가 너무 어려웠어서 골드랑 실버가 수준이 비슷한 것 같다.

 

이진 탐색 코드&방법

아래에 있는 링크에 설명되어 있음

https://seaotter.tistory.com/21

 

[이진 탐색] 이진탐색 방법 & 자바 코드

이진탐색 이진 탐색은 정렬된 리스트에서 검색 범위를 절반씩 줄여가며 탐색하는 알고리즘이다. 시간복잡도: O(log n) 이진탐색 하는 방법 1. 이진탐색할 배열을 정렬한다. 2. 처음에는 배열의 0번

seaotter.tistory.com

 

댓글