이것이 코딩 테스트다 - 그리디(거스름돈, 큰 수의 법칙)

최근 알고리즘 문제를 무작정 풀고 모르면 답을 보고 다음에도 비슷한 유형이 나오면 같은 방식으로 풀고... 하는 식으로 하다가 문제 유형을 먼저 공부해야 한다는 의견을 들어서 알고리즘 관련 서적을 구입했다.

책에 나온 코드는 파이썬으로 되어있지만 깃헙에 자바 코드도 올라와 있어서 참고하면서 풀어보기로 했다.

 

 

그리디

- 단순하지만 강력한 문제 해결 방법
- 현재 상황에서 지금 당장 좋은 것만 고르는 방법
- 매 순간 가장 좋아보이는 것으로 선택하며, 현재의 선택이 나중에 미칠 영향에 대해서는 고려하지 않는다

 

 

 

거스름돈 계산 문제

거스름돈으로 사용할 500원, 100원, 50원, 10원짜리 동전이 무한히 존재한다고 가정한다.
손님에게 거슬러 주어야 할 돈이 N원일 때 거슬러 주어야 할 동전의 최소 개수를 구하자.

public class Main {
	public static void main(String[] args) throws Exception{
		Scanner in = new Scanner(System.in);
		int[] coin =  {500, 100, 50, 10}; //동전의 종류
		int N = in.nextInt(); //받는 돈
		int count = 0; //동전 개수
		for(int i=0; i<coin.length; i++) {
			count+= N / coin[i]; //큰 수부터 나눈 값을 동전 개수에 더한다
			N %= coin[i]; //나머지를 다시 N에 대입한다
		}
		System.out.println(count);
	}
}

 

 

큰 수의 법칙

다양한 수로 이루어진 배열이 있을 때 주어진 수들을 M번 더해서 가장 큰 수를 만드는 법칙이 있다.

단, 배열의 특정한 인덱스에 해당하는 수가 연속해서 K번을 초과하여 더해질 수는 없다.

단, 서로 다른 인덱스에 해당하는 수가 같은 경우에도 서로 다른 것으로 간주한다.

배열의 크기 N, 숫자가 더해지는 횟수 M, 그리고 K가 주어질 때 큰 수의 법칙에 따른 결과를 출력하시오.

 

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()); 
	     //StringTokenizer : 띄어쓰기 기준으로 문자열 분리
	     int N = Integer.parseInt(st.nextToken());
	     int M = Integer.parseInt(st.nextToken());
	     int K = Integer.parseInt(st.nextToken());
	     
	     int[] arr = new int[N]; //변수들을 담을 배열
	     
	     for(int i=0; i<N; i++) {
	    	 arr[i] = Integer.parseInt(st.nextToken());
	     }
	     Arrays.sort(arr); //오름차순 정렬
		
	     int big = arr[N-1]; //가장 큰 수
	     int big2 = arr[N-2]; //두번째로 큰 수
	     int result = 0;
	     while(M != 0) { //M번만큼 반복
	    	 for(int i = 0; i < K; i++) {
	    		 result += big; //가장 큰 수 K번만큼 더함
	    		 M--;
	    	 }
	    	 result += big2; //두번째로 큰 수 더함
	    	 M--;
	     }
	     System.out.println(result);
	}
}

 

답을 보고서도 고통받다가 팍 이해하게 됨 ㅋㅋㅋㅋㅋ ㅠㅠㅠ 이해하고 나니까 정말 별거없었다...

어차피 가장 큰 수랑 두번째 큰 수 외에는 쓰지 않는다.

가장 큰 수를 K번만큼 더하고, 거기에 두번째 큰 수를 한번 더한 것을 한 묶음으로 생각하여 반복을 돌리면 되는 것이다.