Cod sursa(job #437622)

Utilizator sdnxptVasiliu Radu sdnxpt Data 9 aprilie 2010 23:15:31
Problema Gutui Scor 100
Compilator cpp Status done
Runda teme_upb Marime 2.31 kb
#include <fstream>
#include <vector>
#include <algorithm>
#include <queue>

using namespace std;

class Gutuie{
    public:
        unsigned int greutate, nivel;
};

bool comp1(Gutuie a, Gutuie b){
    if( a.nivel < b.nivel )
        return false;
    else
        if( a.nivel == b.nivel ){
            if( a.greutate > b.greutate )
                return true;
            else
                return false;
        }
        else
            return true;
}

bool comp2(unsigned int a, unsigned int b){
    return a > b;
}

inline unsigned int nivel( unsigned int a, unsigned int u, unsigned int hmax ){
    return a % u <= hmax % u ? a / u : a / u + 1;
}

int main(){
    unsigned int n, u, hmax;
    ifstream fin("gutui.in");
    ofstream fout("gutui.out");
    fin >> n >> hmax >> u;
    unsigned int i, x;
    vector<Gutuie> v(n);
    for(i = 0; i < n; i++){
        fin >> x >> v[i].greutate;
        v[i].nivel = nivel(x, u, hmax);
    }
    sort( v.begin(), v.end(), comp1 );
    unsigned int nivmax = nivel(hmax, u, hmax), niv = v[0].nivel, nr = 0;
    unsigned int gmax = 0;
    //vector<unsigned int> greutate;
    priority_queue< int, vector<int>, greater<int> > greutate;
    for(i = 0; i < n; i++){
        if( v[i].nivel == niv ){
            if( nr < nivmax - niv + 1 ){
                nr ++;
                if( greutate.size() < nivmax - niv + 1 ){
                    gmax += v[i].greutate;
                    /*greutate.push_back( v[i].greutate );
                    push_heap( greutate.begin(), greutate.end(), comp2 );*/
                    greutate.push( v[i].greutate );
                }
                else
                    if( greutate.top() < v[i].greutate ){
                        gmax -= greutate.top();
                        gmax += v[i].greutate;
                        greutate.pop();
                        greutate.push( v[i].greutate );
                        /*pop_heap( greutate.begin(), greutate.end(), comp2 );
                        greutate.pop_back();
                        greutate.push_back( v[i].greutate );
                        push_heap( greutate.begin(), greutate.end(), comp2 );*/
                    }
            }
        }
        else{
            niv = v[i].nivel;
            nr = 0;
            i --;
        }
    }
    fout << gmax;
    fin.close();
    fout.close();
    return 0;
}