Algorithm/Algorithm 문제
백준_1744 수 묶기 (그리디)
kjyyjk
2024. 1. 29. 14:19
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.Queue;
public class BJ_1744_수묶기 {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
Queue<Integer> queueP; // 양수를 담을 우선순위큐 (내림차순)
Queue<Integer> queueM; // 음수를 담을 우선순이큐 (오름차순)
queueP = new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2 - o1;
}
});
queueM = new PriorityQueue<>();
int cnt0 = 0; // 0 개수
int cnt1 = 0; // 1 개수
for (int i=0; i<n; i++) {
int inputNum = Integer.parseInt(br.readLine());
if (inputNum > 1) {
queueP.add(inputNum);
} else if (inputNum < 0) {
queueM.add(inputNum);
} else if (inputNum == 0) {
cnt0++;
} else {
cnt1++;
}
}
int result = 0;
while (queueP.size() > 1) {
result += queueP.poll() * queueP.poll();
}
if (!queueP.isEmpty()) {
result += queueP.poll();
}
while (queueM.size() > 1) {
result += queueM.poll() * queueM.poll();
}
if (!queueM.isEmpty() && cnt0 == 0) {
result += queueM.poll();
}
result += cnt1;
System.out.println(new StringBuilder().append(result));
}
}
1744번: 수 묶기
길이가 N인 수열이 주어졌을 때, 그 수열의 합을 구하려고 한다. 하지만, 그냥 그 수열의 합을 모두 더해서 구하는 것이 아니라, 수열의 두 수를 묶으려고 한다. 어떤 수를 묶으려고 할 때, 위치에
www.acmicpc.net