Cod sursa(job #2975033)

Utilizator adelina_15InfoAdelina Radoi adelina_15Info Data 5 februarie 2023 10:36:19
Problema Lupul Urias si Rau Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.89 kb
#include <iostream>
#include <fstream>
#include <queue>

using namespace std;

ifstream fin("sao.in");
ofstream fout("sao.out");

int n, k;
long long hp;

priority_queue<int> mons;

int main()
{
    fin >> n >> k >> hp;
    for(int i = 0; i < n; i++)
    {
        int x;
        fin >> x;
        if(x < 0)
            hp -= x;
        else
        {
            hp -= x;
            mons.push(x);
            while(hp <= 0 && !mons.empty() && k > 0)
            {
                k--;
                int adaug = mons.top();
                hp += adaug;
                mons.pop();
            }
            if(hp == 0)
            {
                fout << i+1;
                return 0;
            }
            else if(hp < 0)
            {
                fout << i;
                return 0;
            }

        }
    }
    fout << n;
    return 0;
}