Cod sursa(job #1450409)

Utilizator delia_99Delia Draghici delia_99 Data 13 iunie 2015 11:51:06
Problema Peste Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.02 kb
#include <fstream>
#include <algorithm>
#include <queue>
using namespace std;

struct aww
{
    int fish,t;
};

int n,k,T,i,Max,s,j;
int aux[1010];
long long D[50010];
aww v[50010];

priority_queue < int, vector <int>, greater <int> >H;

inline bool cmp(aww A,aww B)
{
    if(A.t<B.t)
        return true;
    if(A.t==B.t && A.fish<B.fish)
        return true;
    return false;
}

int main()
{
    ifstream f("peste.in");
    ofstream g("peste.out");
    f>>n>>k>>T;
    for(i=1; i<=n; ++i)
    {
        f>>v[i].fish>>v[i].t;
        if(v[i].t>Max)
            Max=v[i].t;
    }
    sort(v+1,v+n+1,cmp);
    for(i=1; i<=n; ++i)
    {
        H.push(v[i].fish);
        s+=v[i].fish;
        if(i>k)
        {
            s-=H.top();
            H.pop();
        }
        aux[v[i].t]=s;
    }
    for(i=1; i<=Max; ++i)
        aux[i]=max(aux[i-1],aux[i]);
    for(i=1; i<=T; ++i)
        for(j=1; j<=i && j<=Max; ++j )
            D[i]=max(D[i],D[i-j]+aux[j]);
    g<<D[T];
    return 0;
}