5-F 질문있습니다 :)

안녕하세요 유태님 ㅎㅎ

전반적으로 잘 짜셨는데 미묘한 차이가 있습니다.

    ll j = 0;
    for (ll i = 0; i < K; i++)
    {
        for (; j < N && vec[j].first <= Size[i]; j++)
        {
            pq.push(vec[j].second);
        }

이부분을 이렇게 짜야 합니다. N보다 작으면서 && 사이즈작을 때 -> push한다 그게아니라면 for문 종료 이렇게 되어야 합니다.

제가 다듬은 전체코드는 다음과 같습니다.

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>

using namespace std;

typedef long long ll;

ll N, K, result;
priority_queue<ll> pq;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);

    cin >> N >> K;
    vector<pair<ll, ll>> vec(N);
    vector<ll> Size(K);

    for (ll i = 0; i < N; i++)
    {
        cin >> vec[i].first >> vec[i].second;
    }
    for (ll i = 0; i < K; i++)
    {
        cin >> Size[i];
    }

    sort(vec.begin(), vec.end());
    sort(Size.begin(), Size.end());

    ll j = 0;
    for (ll i = 0; i < K; i++)
    {
        for (; j < N && vec[j].first <= Size[i]; j++)
        {
            pq.push(vec[j].second);
        }

        if (!pq.empty())
        {
            result += pq.top();
            pq.pop();
        }
    }

    cout << result << '\n';

    return 0;
}

또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.

인프런 커뮤니티 질문&답변