AtCoder CODE THANKS FESTIVAL 2017 C問題 Factory

Question

前提

問題
解説

実現したいこと

以下のコードでWAが出る理由を知りたいです。

該当のソースコード

#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<ll, ll> P;

int main() {
    ll N, K;
    cin >> N >> K;
    set<P> st;
    ll a, b;
    for (ll i = 0; i < N; i++) {
        cin >> a >> b;
        st.insert(P(a, b));
    }
    ll result = 0;
    while (K > 0) {
        auto p = *st.begin();
        st.erase(p);
        result += p.first;
        p.first += p.second;
        st.insert(p);
        K--;
    }
    cout << result << endl;
}

Answer 1

全く同じスペックの機械が複数ある場合、どうなるでしょうか。
例えば、以下のような入力に対し、正しい出力は2ですが、このコードだと3が返ります。

2 2
1 1
1 1

Comments

[a9uaDrops]

回答ありがとうございます。
setではなくmultisetを使用したらACできました。