[Algorithm] 탐욕(그리디) 알고리즘 (greedy algorithm)

 

 

그리디 알고리즘은 문제를 해결하는 과정에서 가장 좋다고 생각하는 방법을 선택하는 알고리즘이다.

 

 

 

 

 

예를 들어

 

 

 

 

노드 값의 최대 합은 5 -> 7 -> 9를 거쳐 21임을 알 수 있다.

 

 

 

그러나 그리디 알고리즘을 사용하면 그때그때 최대값이 될 것이라고 추측되는 값을 사용하게 된다.

 

 

 

 

 

 

따라서 5 다음에 7, 10, 8 중 가장 큰 값인 10을 사용하고 4, 3 중 큰 값인 4를 사용하게 된다.

 

 

 

이처럼 그리디 알고리즘이 문제의 해결에 있어서 최적의 해를 매번 보장하지는 않는다.

 

 

 

 

그러나 그리디 알고리즘은 계산 속도가 빠르다. 그리디 알고리즘이 적절하다고 판단되는 문제에서는 최적의 해를 빠르게 산출할 수 있다.

 

 

 

 

 

 

 

예시

출처 : 이것이 취업을 위한 코딩 테스트다

 

 

 

 

예를 들어 N = 5이고 각 모험가의 공포도가 2 3 1 2 2 라고 가정하면

 

공포도가 1, 2, 3인 그룹 1과 2, 2인 그룹2로 총 2개의 그룹을 만들 수 있다.

 

+ 모든 모험가를 사용하지 않아도 된다.

 

 

 

 

 

 

 

 

 

 

 

다음과 같이 문제가 주어졌을 때 생각해야 할 첫번째는 오름차순 정렬이다.

 

2 3 1 2 2 -> 1 2 2 2 3 으로 보기 좋게 정렬해준다.

 

 

직관적으로 생각했을 때 공포도가 1인 사람은 혼자서 팀을 구성할 수 있고

 

2인 사람은 다음 사람의 직관도가 2라면 2명이서 구성, 3이라면 다음다음 사람의 직관도를 확인하는 식으로 흘러간다.

 

 

 

 

그러므로 그룹 인원수를 count하고 자리 위치에 따라 공포도의 숫자를 확인하면 된다.

 

그룹 인원수 >= 확인한 자리 사람의 공포도 가 될 때 이를 그룹으로 선택하고 result group에 +1을 해주면 된다. (count=0으로 초기화)

 

 

 

 

 

 

코드

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.StringTokenizer;

public class Greedy {
	
	public static int n;
	public static ArrayList<Integer> arraylist = new ArrayList<>();
	
	public static void main(String[] args) throws IOException {
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int N = Integer.parseInt(br.readLine());
		
		StringTokenizer st = new StringTokenizer(br.readLine(), " ");
		
		for(int i=0; i<N; i++)
			arraylist.add(i, Integer.parseInt(st.nextToken()));
		
		Collections.sort(arraylist);
		
		int result = 0; // 그룹 수
		int count = 0; // 그룹 인원수
		for (int i = 0; i < N; i++) {
			count += 1;
			if(count >= arraylist.get(i)) {
				result += 1;
				count = 0;
			}
		}
		
		System.out.println(result);
	}
}

 

 

책의 코드는 scanner를 사용하였지만 나는 BufferedReader를 사용하였다.

 

 

 

오름차순 정렬을 할 수 있는 방법은 arraylist를 사용하거나

 

그냥 배열을 쓰고싶을 경우 java.util.Arrays 클래스를 사용하여 Arrays.sort(array); 해주면 된다.

 

 

 

 

 

 

 

 

출처 : https://www.youtube.com/watch?v=2zjoKjt97vQ&list=PLRx0vPvlEmdAghTr5mXQxGpHjWqSz0dgC&index=2