IM

작지만 단단하게

Sorting and Searching (결정알고리즘)

뮤직비디오(결정알고리즘)

  • 설명 : 지니레코드에서는 불세출의 가수 조영필의 라이브 동영상을 DVD로 만들어 판매하려 한다. DVD에는 총 N개의 곡이 들어가는데, DVD에 녹화할 때에는 라이브에서의 순서가 그대로 유지되어야 한다. 순서가 바뀌는 것을 우리의 가수 조영필씨가 매우 싫어한다. 즉, 1번 노래와 5번 노래를 같은 DVD에 녹화하기 위해서는 1번과 5번 사이의 모든 노래도 같은 DVD에 녹화해야 한다. 또한 한 노래를 쪼개서 두 개의 DVD에 녹화하면 안된다. 지니레코드 입장에서는 이 DVD가 팔릴 것인지 확신할 수 없기 때문에 이 사업에 낭비되는 DVD를 가급적 줄이려고 한다. 고민 끝에 지니레코드는 M개의 DVD에 모든 동영상을 녹화하기로 하였다. 이 때 DVD의 크기(녹화 가능한 길이)를 최소로 하려고 한다. 그리고 M개의 DVD는 모두 같은 크기여야 제조원가가 적게 들기 때문에 꼭 같은 크기로 해야 한다.

  • 입력 : 첫째 줄에 자연수 N(1≤N≤1,000), M(1≤M≤N)이 주어진다. 다음 줄에는 조영필이 라이브에서 부른 순서대로 부른 곡의 길이가 분 단위로(자연수) 주어진다. 부른 곡의 길이는 10,000분을 넘지 않는다고 가정하자.

  • 출력 : 첫 번째 줄부터 DVD의 최소 용량 크기를 출력하세요.

  • 풀이 답안

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Main t = new Main();
        Scanner scan = new Scanner(System.in);
        int size = scan.nextInt();
        int cd = scan.nextInt();
        int[] arr = new int[size];
        for(int i=0; i<size; i++){
            arr[i] = scan.nextInt();
        }

        System.out.println(t.solution(arr, cd));

    }

    public int count(int[] arr, int capacity){
        int cnt = 1, sum=0;
        for(int x : arr){
            if(sum+x<=capacity){
                sum+=x;
            } else {
                cnt++;
                sum=x;
            }
        }
        return cnt;
    }

    public int solution(int[] arr, int cd){
        int result = 0;

        int lt=Arrays.stream(arr).max().getAsInt(), rt=Arrays.stream(arr).sum();
        while(lt<=rt){
            int mid=(lt+rt)/2;
            if(count(arr, mid) <= cd){
                result=mid;
                rt=mid-1;
            } else {
                lt=mid+1;
            }
        }
        return result;
    }
}
  • 정리
    1. 찾고자 하는 값이 데이터의 예상범위 안에서 출력될때에는 결정 알고리즘을 사용
    2. 일단 cd에 데이터를 담으려면 가장 큰 값이 들어갈수있는 사이즈여야하고, 최대 전체가 다 들어갈 가능성이 있기에 lt는 배열중 최대데이터, rt는 배열의 전체 합으로 지정
    3. 이때 for문을 돌려 배열 내부값을 찾는것이 아니라 Arrays.stream()를 이용함
      • Arrays.stream(arr).max().getAsInt() : 배열의 최대값. 리턴형식을 위해 max()뒤에 꼭 .getAsInt()를 해줘야함. 최소값의 경우도 똑같음
      • Arrays.stream(arr).sum() : 배열원소의 합. int형식으로 리턴되기에 .getAsInt()를 안해도 괜찮음
    4. 반복을 돌아서 중앙점을 찾고, 그 중앙값을 cd크기로 하면 몇개의 cd가 필요한지 구함 => count 함수를 따로 만듬
    5. 원소를 반복으로 돌며 mid값을 초과할때마다 cd 갯수를 증가함
    6. cd갯수가 원하는 cd개수를 초과하지 않는다면 rt를 mid-1값으로 바꾸어 더 작은 값으로도 가능한지 체크
    7. cd갯수가 원하는 cd개수를 초과하면 lt를 mid+1값으로 바꾸어 더 큰값중에 찾음

Updated: