본문 바로가기

Dot Algo∙ DS/PS

[BOJ] 백준 1365번 꼬인 전깃줄 (Java)

    #1365 꼬인 전깃줄

    난이도 : 골드 2

    유형 : LIS / 이진탐색 

     

    1365번: 꼬인 전깃줄

    첫 줄에 전봇대의 개수 N(1 ≤ N ≤ 100,000)이 주어지고, 이어서 N보다 작거나 같은 자연수가 N개 주어진다. i번째 줄에 입력되는 자연수는 길 왼쪽에 i번째 전봇대와 연결된 길 오른편의 전봇대가

    www.acmicpc.net

    ▸ 문제

    공화국에 있는 유스타운 시에서는 길을 사이에 두고 전봇대가 아래와 같이 두 줄로 늘어서 있다. 그리고 길 왼편과 길 오른편의 전봇대는 하나의 전선으로 연결되어 있다. 어떤 전봇대도 두 개 이상의 다른 전봇대와 연결되어 있지는 않다.

     

    문제는 이 두 전봇대 사이에 있는 전깃줄이 매우 꼬여 있다는 점이다. 꼬여있는 전깃줄은 화재를 유발할 가능성이 있기 때문에 유스타운 시의 시장 임한수는 전격적으로 이 문제를 해결하기로 했다.

     

    임한수는 꼬여 있는 전깃줄 중 몇 개를 적절히 잘라 내어 이 문제를 해결하기로 했다. 하지만 이미 설치해 놓은 전선이 아깝기 때문에 잘라내는 전선을 최소로 하여 꼬여 있는 전선이 하나도 없게 만들려고 한다.

     

    유스타운 시의 시장 임한수를 도와 잘라내야 할 전선의 최소 개수를 구하는 프로그램을 작성하시오.

     입력

    첫 줄에 전봇대의 개수 N(1 ≤ N ≤ 100,000)이 주어지고, 이어서 N보다 작거나 같은 자연수가 N개 주어진다. i번째 줄에 입력되는 자연수는 길 왼쪽에 i번째 전봇대와 연결된 길 오른편의 전봇대가 몇 번 전봇대인지를 나타낸다.

     출력

    전선이 꼬이지 않으려면 최소 몇 개의 전선을 잘라내야 하는 지를 첫째 줄에 출력한다.

     

    문제 풀이  

    LIS 유형의 문제이다. DP만 사용하면 O(n^2)이지만 인덱스 찾는 과정을 이진탐색을 이용해서 구하면 O(n*logn)에 해결할 수 있다. 최소의 전선을 자르기 위해서는 그냥 가장 긴 증가하는 부분 수열을 구한 다음 거기에 해당하지 않는 전선들을 제거해주면 된다.

    • 즉, ( 전체 전선의 갯수 - LIS 길이 )를 구해주면 된다.

     

    설계

    1. 전깃줄에 대한 수열을 입력받는다.
    2. 해당 수열에 대한 LIS를 이진탐색을 통해 구한다.
      1. if(arr[i] > dp[len]) i번째 수열의 값이 현재 LIS의 최댓값(dp[len])보다 크다면 최댓값을 갱신한다.
        1. dp[len++] = arr[i];
      2. 만약 i번째 수열의 값이 최댓값과 최솟값의 사이에 있다면 이진탐색을 통해 해당 위치를 구하여 갱신해준다.
        1. idx = binarySearch(0, len, arr[i]);
        2. dp[idx] = arr[i];

     

    풀이 코드 

    import java.io.*;
    import java.util.*;
    
    public class Main {
    
    	static int[] dp;
    	public static void main(String[] args) throws IOException{
    		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    		int n = Integer.parseInt(br.readLine());
    		
    		int[] arr = new int[n];
    		StringTokenizer st = new StringTokenizer(br.readLine());
    		for(int i=0; i<n; i++) {
    			arr[i] = Integer.parseInt(st.nextToken());
    		}
    		
    		dp = new int[n+1];
    		int len = 0;
    		int idx = 0;
    		for(int i=0; i<n; i++) {
    			if(arr[i] > dp[len]) {
    				len += 1;
    				dp[len] = arr[i];
    				continue;
    			} 
    			
    			idx = binarySearch(0, len, arr[i]);
    			dp[idx] = arr[i];
    		}
    		
    		System.out.println(n-len);
    				
    				
    	}
    	static int binarySearch(int left, int right, int key) {
    		while(left<right) {
    			int mid = (left+right)/2;
    			if(dp[mid] > key) {
    				right = mid;
    			}else {
    				left = mid+1;
    			}
    		}
    		return right;
    	}
    }