[백준/Baekjoon] 15903 카드 합체 놀이 C++ :: Data structure

https://www.acmicpc.net/problem/15903

15903번: 카드 합체 놀이

---

1. Logic

- 카드를 최소한의 점수로 합체하기 위해서는 가장 작은 값끼리 묶어서 합쳐야 하기 때문에 우선순위 큐를 사용해서 가장 앞에 오는 두개의 수를 가지고 계산을 돌려준다!

 

---

2. Code

#include<bits/stdc++.h>

using namespace std;

priority_queue<long long, vector<long long>, greater<>> pq;

int main() {
    long long n, cnt;
    cin >> n >> cnt;

    for(int i = 0; i < n; i++) {
        int a;
        cin >> a;
        pq.push(a);
    }

    for(int i = 0; i < cnt; i++) {
        long long fir = pq.top();
        pq.pop();
        long long sec = pq.top();
        pq.pop();

        long long sum = fir + sec;

        pq.push(sum);
        pq.push(sum);
    }

    long long ans = 0;
    while(!pq.empty()) {
        ans += pq.top();
        pq.pop();
    }
    cout << ans;
}

---

 

 

3. Feedback

- 정답의 범위를 잘 체크해서 answer 변수 타입을 잘 맞춰주자 

728x90