TopCoder SRM 502 Division 2 250 pts.

SRM_502_Div2_250.cpp

#include <vector>
#include <algorithm>

using namespace std;

class TheProgrammingContestDivTwo {
 public:
  vector<int> find(int T, vector<int> requiredTime) {
    vector<int> res;
    
    int solved  = 0;
    int penalty = 0;
    int remaining = T;
    int elapsed   = 0;
    
    sort(requiredTime.begin(), requiredTime.end());
    
    for(auto p : requiredTime) {
      if( p <= remaining ) {
        remaining -= p;
        solved++;
        penalty += elapsed += p;
      }
    }
    
    res.push_back(solved);
    res.push_back(penalty);
    
    return res;
  }
};