leoanjos

Hash Code 2022

Feb 24th, 2022
208
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 4.79 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4.  
  5. #define long long long int
  6.  
  7. struct Person {
  8.     string name;
  9.     map<string, int> skills;
  10.  
  11.     Person() {}
  12.     Person(string name): name(name) {}
  13. };
  14.  
  15. struct Project {
  16.     string name;
  17.     int days, score, best_before;
  18.     vector<pair<string, int>> roles;
  19.  
  20.     Project() {}
  21.     Project(string name, int days, int score, int best_before):
  22.         name(name), days(days), score(score), best_before(best_before) {}
  23.  
  24.     bool operator <(const Project &other) {
  25.         /* if (score == other.score)
  26.             return best_before < other.best_before;
  27.  
  28.         return score > other.score; */
  29.  
  30.         /* if (best_before == other.best_before)
  31.             return score > other.score;
  32.  
  33.         return best_before < other.best_before; */
  34.  
  35.         if (score == other.score && days == other.days)
  36.             return best_before < other.best_before;
  37.  
  38.         if (score == other.score)
  39.             return days < other.days;
  40.  
  41.         return score > other.score;
  42.     }
  43. };
  44.  
  45. int main() {
  46.     ios_base::sync_with_stdio(false);
  47.     cin.tie(NULL);
  48.  
  49.     int N, M;
  50.     cin >> N >> M;
  51.  
  52.     vector<Person> people(N);
  53.     map<string, set<pair<int, int>>> skills;
  54.  
  55.     vector<string> all_skills;
  56.     for (int i = 0; i < N; i++) {
  57.         string name;
  58.         int number_skills;
  59.         cin >> name >> number_skills;
  60.  
  61.         people[i] = Person(name);
  62.         for (int j = 0; j < number_skills; j++) {
  63.             string skill; int level;
  64.             cin >> skill >> level;
  65.             people[i].skills[skill] = level;
  66.             skills[skill].emplace(level, i);
  67.             all_skills.push_back(skill);
  68.         }
  69.     }
  70.  
  71.     sort(all_skills.begin(), all_skills.end());
  72.     all_skills.erase(unique(all_skills.begin(), all_skills.end()), all_skills.end());
  73.  
  74.     for (int i = 0; i < N; i++)
  75.         for (auto skill: all_skills)
  76.             if (!people[i].skills.count(skill))
  77.                 skills[skill].emplace(0, i);
  78.  
  79.     vector<Project> projects(M);
  80.     for (int i = 0; i < M; i++) {
  81.         string name;
  82.         int days, score, best_before, number_roles;
  83.         cin >> name >> days >> score >> best_before >> number_roles;
  84.  
  85.         projects[i] = Project(name, days, score, best_before);
  86.         for (int j = 0; j < number_roles; j++) {
  87.             string skill; int level;
  88.             cin >> skill >> level;
  89.             projects[i].roles.emplace_back(skill, level);
  90.         }
  91.     }
  92.  
  93.     sort(projects.begin(), projects.end());
  94.  
  95.     int time = 0, score = 0;
  96.     vector<int> free(N, 0);
  97.  
  98.     vector<bool> done(M, false);
  99.     vector<pair<int, vector<int>>> ans;
  100.  
  101.     for (int k = 0; k < 100; k++) {
  102.         for (int i = 0; i < M; i++) {
  103.             if (done[i]) continue;
  104.  
  105.             set<int> used;
  106.             vector<int> curr;
  107.             map<string, int> seen_skills;
  108.  
  109.             bool possible = true;
  110.             for (auto [skill, level]: projects[i].roles) {
  111.                 int tests = 0;
  112.                 if (seen_skills[skill] >= level) level--;
  113.                 auto it = skills[skill].lower_bound(make_pair(level, 0));
  114.  
  115.                 possible = false;
  116.                 while (it != skills[skill].end() && tests < 10 && !possible) {
  117.                     if (used.count(it->second) || free[it->second] > time) it++;
  118.                     else possible = true;
  119.                 }
  120.  
  121.                 if (!possible) break;
  122.  
  123.                 used.insert(it->second);
  124.                 curr.push_back(it->second);
  125.  
  126.                 for (auto [person_skill, person_level]: people[it->second].skills)
  127.                     seen_skills[person_skill] = max(seen_skills[person_skill], person_level);
  128.             }
  129.  
  130.             if (!possible) {
  131.                 time++;
  132.                 continue;
  133.             }
  134.  
  135.             done[i] = true;
  136.             ans.emplace_back(i, curr);
  137.             score += max(projects[i].score - max(time + projects[i].days - projects[i].best_before + 1, 0), 0);
  138.  
  139.             for (int j = 0; j < (int) curr.size(); j++) {
  140.                 int id = curr[j];
  141.                 free[id] = time + projects[i].days;
  142.  
  143.                 auto [skill, level] = projects[i].roles[j];
  144.                 if (people[id].skills[skill] == level) {
  145.                     skills[skill].erase(make_pair(people[id].skills[skill], id));
  146.                     skills[skill].emplace(++people[id].skills[skill], id);
  147.                 }
  148.             }
  149.  
  150.             time = *min(free.begin(), free.end());
  151.         }
  152.     }
  153.  
  154.     cout << ((int) ans.size()) << "\n";
  155.     for (auto [project, roles]: ans) {
  156.         cout << projects[project].name << "\n";
  157.         for (int i = 0; i < (int) roles.size(); i++) {
  158.             if (i) cout << " ";
  159.             cout << people[roles[i]].name;
  160.         }
  161.  
  162.         cout << "\n";
  163.     }
  164.  
  165.     cout << score << "\n";
  166. }
Advertisement
Add Comment
Please, Sign In to add comment