[백준] 17266 어두운 굴다리 자바 풀이 이분탐색 (제출시 7%에서 틀린다면)

[백준] 17266 어두운 굴다리 자바 풀이 이진탐색 (제출시 7%에서 틀린다면)

 

 

 

 

얼마전 소프티어에서 풀었던 녀석과 해결방법이 거의 동일하니

아래의 문제도 풀어보시면 좋을 것 같습니다.

 

2024.08.28 - [알고리즘 단련장/소프티어] - [소프티어] 슈퍼컴퓨터 클러스터 자바 풀이 이진탐색 시간초과 해결 (HSAT 4회 정기 코딩 인증평가 기출)

 

 

 

 

 

 

 

문제링크

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

 

 

 

 

 

 

문제

인하대학교 후문 뒤쪽에는 어두운 굴다리가 있다. 겁쟁이 상빈이는 길이 조금이라도 어둡다면 가지 않는다. 따라서 굴다리로 가면 최단거리로 집까지 갈수 있지만, 굴다리는 어둡기 때문에 빙빙 돌아서 집으로 간다. 안타깝게 여긴 인식이는 굴다리 모든 길 0~N을 밝히게 가로등을 설치해 달라고 인천광역시에 민원을 넣었다. 인천광역시에서 가로등을 설치할 개수 M과 각 가로등의 위치 x들의 결정을 끝냈다. 그리고 각 가로등은 높이만큼 주위를 비출 수 있다. 하지만 갑자기 예산이 부족해진 인천광역시는 가로등의 높이가 높을수록 가격이 비싸지기 때문에 최소한의 높이로 굴다리 모든 길 0~N을 밝히고자 한다. 최소한의 예산이 들 높이를 구하자. 단 가로등은 모두 높이가 같아야 하고, 정수이다.

다음 그림을 보자.

 

다음 그림은 예제1에 대한 설명이다.

 

가로등의 높이가 1일 경우 0~1사이의 길이 어둡기 때문에 상빈이는 지나가지 못한다.

아래 그림처럼 높이가 2일 경우 0~5의 모든 길이 밝기 때문에 상빈이는 지나갈 수 있다.

 

입력

첫 번째 줄에 굴다리의 길이 N 이 주어진다. (1 ≤ N ≤ 100,000)

두 번째 줄에 가로등의 개수 M 이 주어진다. (1 ≤ M ≤ N)

다음 줄에 M 개의 설치할 수 있는 가로등의 위치 x 가 주어진다. (0 ≤ x ≤ N)

가로등의 위치 x는 오름차순으로 입력받으며 가로등의 위치는 중복되지 않으며, 정수이다.

출력

굴다리의 길이 N을 모두 비추기 위한 가로등의 최소 높이를 출력한다.

예제 입력 1 복사

5
2
2 4

예제 출력 1 복사

2

예제 입력 2 복사

3
1
0

예제 출력 2 복사

3

출처

University > 인하대학교 > 2019 인하대학교 프로그래밍 경진대회(IUPC) D번

• 빠진 조건을 찾은 사람: kipa00
• 문제의 오타를 찾은 사람: lekonell
• 문제를 만든 사람: san9407

알고리즘 분류

• 구현
• 이분 탐색

 

 

 

 

 

 

핵심 코드

이분탐색으로 특정 높이에 대해 모든 길을 비출 수 있는지 체크하는 로직에서

"가장 마지막 스탠드의 우측 영역이 길이N까지 비추는지" 확인해야 합니다.

 

이 부분을 해결하지 않는다면 백준 제출시,

7%까지 맞추다가 틀렸다고 나올겁니다.

// 히든 테스트 케이스
// 마지막 전등의 우측 offSet이, 길이n까지 못비춘다면
if(leftPos < n){    // if(stand[m-1]+h < n){ 와 동일조건
    return false;
}

 

 

 

전체 코드

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

public class Main {

    public static long dap = 0;
    public static long n;
    public static int m;

    public static void main(String[] args) throws IOException{

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        n = Integer.parseInt(br.readLine());    // 굴다리 길이
        m = Integer.parseInt(br.readLine());    // 가로등 개수

        StringTokenizer st = new StringTokenizer(br.readLine());
        Integer stand[] = new Integer[m];
        for(int i=0; i<m; i++){
            stand[i] = Integer.parseInt(st.nextToken());
        }

        // Arrays.sort(arr);
        int height = 1;
        long left = 0;
        long right = n;
        long mid = n/2;


        while(left <= right){
            mid = (left + right) / 2;
            boolean canLight = binary_search(stand, mid);
            // System.out.println("L, R, mid, YN : " + left + ", " + right + ", " + mid + ", " + canLight);
            if(canLight){   // 전부 비출 수 있네..전등 높이가 너무 높나? (바이너리 좌측영역으로 이동)
                dap = mid;
                right = mid-1;
            }else{  // 전부 못비춰. 높이 더 높아야해 (바이너리 우측영역으로 이동)
                left = mid+1;
            }
        }

        if(m == 1){
            dap = n;
        }

        bw.write(String.valueOf(dap) + "\n");
        bw.flush();
        bw.close();
    }

    public static boolean binary_search(Integer stand[], long h){
        // stand 위치마다 h높이 전등을 세웠을때, N길이의 모든 영역을 비출 수 있나요?
        long leftPos = 0;
        for(int i=0; i<stand.length; i++){
            long leftLightOffset = stand[i]-h;
            if(leftLightOffset <= leftPos){    // 왼쪽 비추기 불가능
                leftPos = stand[i]+h;
            }else{
                return false;
            }
        }

        // 히든 테스트 케이스
        // 마지막 전등의 우측 offSet이, 길이n까지 못비춘다면
        if(leftPos < n){    // if(stand[m-1]+h < n){ 와 동일조건
            return false;
        }
        return true;
    }
}