Cod sursa(job #1956518)

Utilizator Daria09Florea Daria Daria09 Data 6 aprilie 2017 20:03:58
Problema Peste Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <iostream>
#include <fstream>
#define DIM 50001
#include <algorithm>
using namespace std;
ifstream f("peste.in");
ofstream g("peste.out");
struct peste
{
    int p,t;
}a[DIM];
int timp,n,k,d[DIM],cost[DIM],sol[DIM];
void read()
{
    f>>n>>k>>timp;
    for(int i=1;i<=n;i++)
        f>>a[i].p>>a[i].t;
    f.close();
}
inline bool cmp(peste a,peste b){if(a.t==b.t)return a.p<b.p; return a.t<b.t;}
void solve()
{
    sort(a+1,a+n+1,cmp);
    int j=1,s=0,head=1,tail=0;
    for(int i=1;i<=DIM-1;i++)
    {
        while(j<=n&&a[j].t==i)
        {
            s+=a[j].p; d[++tail]=a[j].p;
            if(tail>k){s-=d[head];++head;}
            ++j;
        }
        cost[i]=s;
    }
    for(int i=1;i<=timp;i++)
        for(j=i-1;j>max(-1,i-DIM);j--)
        sol[i]=max(sol[i],sol[j]+cost[i-j]);
    g<<sol[timp];
}
int main()
{
    read();
    solve();
    return 0;
}