백준 1781번 컵라면

 Description

상욱 조교는 동호에게 N개의 문제를 주고서, 각각의 문제를 풀었을 때 컵라면을 몇 개 줄 것인지 제시 하였다. 하지만 동호의 찌를듯한 자신감에 소심한 상욱 조교는 각각의 문제에 대해 데드라인을 정하였다.

문제 번호	1	2	3	4	5	6	7
데드라인	1	1	3	3	2	2	6
컵라면 수	6	7	2	1	4	5	1

위와 같은 상황에서 동호가 2, 6, 3, 1, 7, 5, 4 순으로 숙제를 한다면 2, 6, 3, 7번 문제를 시간 내에 풀어 총 15개의 컵라면을 받을 수 있다.

문제는 동호가 받을 수 있는 최대 컵라면 수를 구하는 것이다. 위의 예에서는 15가 최대이다.

문제를 푸는데는 단위 시간 1이 걸리며, 각 문제의 데드라인은 N이하의 자연수이다., 각 문제를 풀 때 받을 수 있는 컵라면 수와 최대로 받을 수 있는 컵라면 수는 모두 231보다 작거나 같은 자연수이다.

 Input

첫 줄에 숙제의 개수 N (1 ≤ N ≤ 200,000)이 들어온다. 다음 줄부터 N+1번째 줄까지 i+1번째 줄에 i번째 문제에 대한 데드라인과 풀면 받을 수 있는 컵라면 수가 공백으로 구분되어 입력된다.

 Output

첫 줄에 동호가 받을 수 있는 최대 컵라면 수를 출력한다.

 Solution

  • 처음에는 이중 배열에서 같은 데드라인을 기준으로 최대의 컵라면 수를 갖고있는 원소들을 더했는데, 통과되지 않는 테스트케이스가 있었다.
  • 생각해보니 위와 같이 코드를 짠다면, 아래의 경우에서 2+3+7=12개의 답을 내게 된다. 그러나 실제로는 7+7+7=21개의 컵라면을 받을 수 있다.
문제번호 1 2 3 4 5 6 7
데드라인 1 1 2 2 3 3 3
컵라면수 1 2 2 3 7 7 7
  • 어제 우선순위 큐를 배웠던 것이 생각나 우선순위 큐를 이용하여 풀었더니 정답 처리를 받았다.
import heapq

n = int(input())
arr = []

for i in range(n):
    data = list(map(int, input().split()))
    arr.append(data)

arr.sort()

answer = []
for i in arr:
    deadline = i[0]
    heapq.heappush(answer, i[1])
    if len(answer) > deadline:
        heapq.heappop(answer)

print(sum(answer))

 Feedback

  • 모범풀이와 비슷하게 풀어서 기분이 좋다.
import heapq

n = int(input())
array = []
q = []

for i in range(n):
    a, b = map(int, input().split(' '))
    array.append((a, b))
array.sort()

for i in array:
    a = i[0]
    heapq.heappush(q, i[1])
    if a < len(q):
        heapq.heappop(q)

print(sum(q))
  • java 풀이도 살펴보았다. 자바 자료구조 사용법을 익히기 좋은 것 같다.
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.Scanner;

class Node {
	public int x;
	public int y;
	Node (int x, int y) {
		this.x = x;
		this.y = y;
	}
}
  • java에서는 여러개의 원소를 연결 리스트 안에 담기 위해 새로운 클래스를 구조체로 사용하나보다. 이 풀이 말고 다른 풀이도 Node, Homework 등의 class를 정의하였다.
public class Main {

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

		int n = sc.nextInt();
		int result = 0;
		ArrayList<Node> data = new ArrayList<Node>();
		PriorityQueue<Integer> pq = new PriorityQueue<Integer>();

		// 문제 정보를 입력 받은 이후에, 데드라인을 기준으로 정렬
		for (int i = 0; i < n; i++) {
			data.add(new Node(sc.nextInt(), sc.nextInt()));
		}

		Collections.sort(data, new Comparator<Node>() {
			@Override
			public int compare(Node a, Node b) {
				if (a.x == b.x) {
					return Integer.compare(a.y, b.y);
				}
				return Integer.compare(a.x, b.x);
			}
		});
		for (int i = 0; i < n; i++) {
			int a = data.get(i).x;
			int b = data.get(i).y;
			pq.offer(b);
			// 데드라인을 초과하는 경우에는 최소 원소를 제거
			if (a < pq.size()) {
				pq.poll();
			}
		}

		while (!pq.isEmpty()) {
			result += pq.poll();
		}

		System.out.println(result);
	}

}

Source