[Baekjoon] 7774_콘센트

문제(출처: https://www.acmicpc.net/problem/7774)

---

< 콘센트 >

 

문제 풀이 

 

우선순위 큐를 사용하여 콘센트가 많은 순으로 꽂는다.

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Collections;
import java.util.PriorityQueue;
import java.util.StringTokenizer;
public class _7774_ { // 콘센트
	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(bf.readLine());
		int n = Integer.parseInt(st.nextToken());
		int m = Integer.parseInt(st.nextToken());
		PriorityQueue<Integer> a = new PriorityQueue<>(Collections.reverseOrder());
		PriorityQueue<Integer> b = new PriorityQueue<>(Collections.reverseOrder());
		st = new StringTokenizer(bf.readLine());
		for (int i = 0; i < n; i++) {
			a.add(Integer.parseInt(st.nextToken()));
		}
		st = new StringTokenizer(bf.readLine());
		for (int i = 0; i < m; i++) {
			b.add(Integer.parseInt(st.nextToken()));
		}
		int result = 0;
		while (!a.isEmpty() && !b.isEmpty()) {
			if (a.size() < n) {
				result -= 1;
			}
			int temp = a.poll();
			while (temp-- > 0 && !b.isEmpty()) {
				result += b.poll();
			}
		}
		System.out.println(result);
	}
}

변수)
n, m : 첫 번째, 두 번째 멀티탭의 개수
a, b : 멀티탭에 있는 콘센트 개수
result : 사용할 수 있는 컴퓨터 개수

 

첫 번째, 두 번째 멀티탭의 개수를 입력받는다. 멀티탭의 개수만큼 콘센트 개수를 입력받아 각 우선순위 큐에 저장하여 내림차순으로 정렬한다. a와 b의 우선순위 큐가 비어있지 않을 때 다음 과정을 반복한다.

1) a 우선순위 큐가 n개보다 작다면 표준 A 콘센트를 하나 사용한 것이므로 result-1

2) a 우선수위 큐 poll

3) 앞에서 poll 한 개수만큼 b 우선순위 큐에서 poll 하여 result에 값 추가

 

최종 result를 출력한다.

---

---