[Ch.06 - SortSearch] 09. 뮤직비디오(결정알고리즘) ##

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

 

설명

지니레코드에서는 불세출의 가수 조영필의 라이브 동영상을 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의 최소 용량 크기를 출력하세요.

예시 입력 1 

9 3
1 2 3 4 5 6 7 8 9

예시 출력 1

17

힌트

설명 : 3개의 DVD용량이 17분짜리이면 (1, 2, 3, 4, 5) (6, 7), (8, 9) 이렇게 3개의 DVD로 녹음을 할 수 있다.

17분 용량보다 작은 용량으로는 3개의 DVD에 모든 영상을 녹화할 수 없다.

---

---

import java.util.Scanner;

public class Main {

	static boolean check(int mid, int[] arr) {
		int sum = 0;
		int cnt = 1;
		boolean flag = true;
		for (int i = 0; i < n; i++) {
			sum += arr[i];
			if (sum > mid) {
				sum = arr[i];
				cnt++;
				if (cnt > m)
					return flag = false;
			}
		}
		return flag;
	}

	static int lt, rt, n, m;

	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		m = in.nextInt();
		int[] arr = new int[n];
		lt = Integer.MIN_VALUE;
		rt = 0;
		for (int i = 0; i < n; i++) {
			arr[i] = in.nextInt();
			lt = Math.max(lt, arr[i]);
			rt += arr[i];
		}
		System.out.println(solution(arr));
	}

	static int solution(int[] arr) {
		int answer = 0;
		while (lt <= rt) {
			int mid = (lt + rt) / 2;
			if (check(mid, arr)) {
				rt = mid - 1;
				answer = mid;
			} else
				lt = mid + 1;
		}
		return answer;
	}
}

 

 

+) 세련된 풀이

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

public class Main {
	public int count(int[] arr, int capacity) {
		// DVD장수와 DVD에 담는 곡들의 합 변수 선언
		int cnt = 1, sum = 0;
		for (int x : arr) {
			if (sum + x > capacity) {
				cnt++;
				sum = x;
			} else
				sum += x;
		}
		return cnt;

	}

	public int solution(int n, int m, int[] arr) {
		// 최소 용량 크기 결정
		// lt=0, rt=총합

		int answer = 0;
		int lt = Arrays.stream(arr).max().getAsInt();
		int rt = Arrays.stream(arr).sum();

		int mid = (lt + rt) / 2;

		// 종료 조건 -> lt는 항상 증가 rt는 감소 -> 범위를 줄인다.
		while (lt <= rt) {
			mid = (lt + rt) / 2;
			if (count(arr, mid) <= m) {
				answer = mid;
				rt = mid - 1;
				
			} else
				lt = mid + 1;

				

		}

		return answer;
	}

	public static void main(String[] args) {
		Main T = new Main();
		Scanner kb = new Scanner(System.in);

		int n = kb.nextInt();
		int m = kb.nextInt();
		int[] arr = new int[n];
		for (int i = 0; i < n; i++)
			arr[i] = kb.nextInt();

		System.out.println(T.solution(n, m, arr));
	}
}

 

 

# arr에서 가장 큰 값 찾기

//1.
int lt = Integer.MIN_VALUE;
for(int i=0;i<n;i++){
    lt = Math.max(lt, arr[i]);
}

//2.
int lt = Arrays.stream(arr).max().getAsInt();

# arr 합계 찾기

//1.
int rt=0;
for(int i=0;i<n;i++){
    rt += arr[i];
}

//2.
int rt = Arrays.stream(arr).sum();

# m장에 mid분이 가능한지 찾는 함수

1. boolean이 반환값

static boolean check(int mid, int[] arr) {
    int sum = 0;
    int cnt = 1;
    boolean flag = true;
    for (int i = 0; i < n; i++) {
        sum += arr[i];
        if (sum > mid) {
            sum = arr[i];
            cnt++;
            if (cnt > m)
                return flag = false;
        }
    }
    return flag;
}


while (lt <= rt) {
    int mid = (lt + rt) / 2;
    if (check(mid, arr)) {
        rt = mid - 1;
        answer = mid;
    } else
        lt = mid + 1;
}

2. 장 수가 반환값

public int count(int[] arr, int capacity) {
    // DVD장수와 DVD에 담는 곡들의 합 변수 선언
    int cnt = 1, sum = 0;
    for (int x : arr) {
        if (sum + x > capacity) {
            cnt++;
            sum = x;
        } else
            sum += x;
    }
    return cnt;

}

while (lt <= rt) {
    mid = (lt + rt) / 2;
    if (count(arr, mid) <= m) {
        answer = mid;
        rt = mid - 1;

    } else
        lt = mid + 1;

}

 

 

3. 마구간 정하기 (결정 알고리즘)

---

최소 최대 정하기

int lt = Arrays.stream(arr).max().getAsInt();
int rt = Arrays.stream(arr).sum();

 

import java.util.*;
  
public class Main {
  public static void main(String[] args){
    Scanner in=new Scanner(System.in);
    int n = in.nextInt();
    int k = in.nextInt();
    int[] arr = new int[n];
    
    int min=Integer.MAX_VALUE;
    int max=Integer.MIN_VALUE;
    for(int i=0;i<n;i++){
      arr[i]=in.nextInt();
      if(min>arr[i]) min=arr[i];
      if(max<arr[i]) max=arr[i];
    }
//    Arrays.sort(arr);
    //N개의 노래를, K개의 DVD에
    int lt = Arrays.stream(arr).max().getAsInt();
	int rt = Arrays.stream(arr).sum();
    //System.out.println("lt "+lt+" "+"rt "+rt);
    int mid=0;
    int answer=0;
    
    while(lt<=rt){
      mid=(lt+rt)/2;
      if(check(arr, n, k, mid)){
        answer=mid;
        rt=mid-1;
        //System.out.println("true "+lt+" "+rt);
      }else{
        lt=mid+1;
        //System.out.println("false "+lt+" "+rt);

      }
    }
    System.out.println(answer);
  }
  private static boolean check(int[] arr, int n, int k, int mid){
    int sum=0;
    int cnt=1;
    for(int i=0;i<n;i++){
      sum+=arr[i];
      if(sum>mid) {
        cnt++;
        sum=arr[i];
      }
      //System.out.println(i+" "+mid+" "+sum+" "+cnt);
      if(cnt>k) return false;
    }

    return true;
  }
}