본문 바로가기

독서/알고리즘

백준 2110 공유기 설치

728x90

이분 탐색을 활용한 매개변수 탐색 알고리즘

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;
import java.util.stream.Collectors;

public class Main {

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine().trim());

        int n = Integer.parseInt(st.nextToken());
        int c = Integer.parseInt(st.nextToken());

        Houses houses = new Houses();
        for (int i = 0; i < n; i++) {
            houses.addHouse(new House(Integer.parseInt(br.readLine().trim())));
        }

        Engineer engineer = new Engineer(c);
        System.out.println(engineer.setup(houses));
    }

    private static class Engineer {

        private final int wifiNum;

        public Engineer(int wifiNum) {
            this.wifiNum = wifiNum;
        }

        public int setup(Houses houses) {
            Houses sortedHouses = houses.sortedHouses();
            int answer = 0;
            int s = 1;
            int e =
                sortedHouses.getHouse(sortedHouses.size() - 1)
                    .distance(sortedHouses.getHouse(0));
            while (s <= e) {
                int mid = (s + e) / 2;
                int setupCount = setup(sortedHouses, mid);
                if (setupCount >= this.wifiNum) {
                    answer = mid;
                    s = mid + 1;
                    continue;
                }
                e = mid - 1;
            }
            return answer;
        }

        private int setup(Houses houses, int gap) {
            int result = 1;
            House base = houses.getHouse(0);
            int len = houses.size();
            for (int i = 1; i < len; i++) {
                if (isPossible(houses.getHouse(i).distance(base), gap)) {
                    result++;
                    base = houses.getHouse(i);
                }
            }
            return result;
        }

        private boolean isPossible(int distance, int gap) {
            return distance >= gap;
        }
    }

    private static class Houses {

        private final List<House> houses = new ArrayList<>();

        public Houses() {

        }

        public Houses(List<House> houses) {
            this.houses = houses;
        }

        public void addHouse(House house) {
            this.houses.add(house);
        }

        public Houses sortedHouses() {
            return new Houses(
                new ArrayList<>(this.houses.stream().sorted().collect(Collectors.toList())));
        }

        public House getHouse(int i) {
            return new House(this.houses.get(i).getPos());
        }

        public int size() {
            return this.houses.size();
        }
    }

    private static class House implements Comparable<House> {

        private final int pos;

        public House(int pos) {
            if (pos < 0) {
                throw new IllegalArgumentException();
            }
            this.pos = pos;
        }

        public int getPos() {
            return pos;
        }

        public int distance(House o) {
            return Math.abs(this.pos - o.pos);
        }

        @Override
        public int compareTo(House o) {
            return this.pos - o.pos;
        }
    }
}

'독서 > 알고리즘' 카테고리의 다른 글

연결 요소의 개수  (0) 2023.12.15
백준 1965 상자넣기  (0) 2022.01.10
백준 23352 방탈출  (0) 2021.12.28
백준 1188 음식 평론가  (0) 2021.12.27
백준 14567 선수과목  (0) 2021.12.27