From d46cc2bf8ef24ecae7974dbcd13ba4f416446969 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=9C=D0=BE=D1=80=D0=B5=D0=B2=20=D0=90=D0=BB=D0=B5=D0=BA?=
 =?UTF-8?q?=D1=81=D0=B0=D0=BD=D0=B4=D1=80?= <hikjik@yandex.ru>
Date: Tue, 17 Oct 2023 21:36:47 +0300
Subject: [PATCH] Minimum Hours of Training to Win a Competition

---
 CMakeLists.txt                                |  1 +
 .../CMakeLists.txt                            |  1 +
 .../solution.hpp                              | 30 +++++++++++++++
 .../test.cpp                                  | 37 +++++++++++++++++++
 4 files changed, 69 insertions(+)
 create mode 100644 solutions/minimum-hours-of-training-to-win-a-competition/CMakeLists.txt
 create mode 100644 solutions/minimum-hours-of-training-to-win-a-competition/solution.hpp
 create mode 100644 solutions/minimum-hours-of-training-to-win-a-competition/test.cpp

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 760f19d1..f9bc6685 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -413,6 +413,7 @@ add_task(minimum-distance-between-bst-nodes)
 add_task(minimum-falling-path-sum)
 add_task(minimum-flips-to-make-a-or-b-equal-to-c)
 add_task(minimum-genetic-mutation)
+add_task(minimum-hours-of-training-to-win-a-competition)
 add_task(minimum-insertion-steps-to-make-a-string-palindrome)
 add_task(minimum-jumps-to-reach-home)
 add_task(minimum-limit-of-balls-in-a-bag)
diff --git a/solutions/minimum-hours-of-training-to-win-a-competition/CMakeLists.txt b/solutions/minimum-hours-of-training-to-win-a-competition/CMakeLists.txt
new file mode 100644
index 00000000..0690dcc7
--- /dev/null
+++ b/solutions/minimum-hours-of-training-to-win-a-competition/CMakeLists.txt
@@ -0,0 +1 @@
+add_catch(test_minimum_hours_of_training_to_win_a_competition test.cpp)
diff --git a/solutions/minimum-hours-of-training-to-win-a-competition/solution.hpp b/solutions/minimum-hours-of-training-to-win-a-competition/solution.hpp
new file mode 100644
index 00000000..86540037
--- /dev/null
+++ b/solutions/minimum-hours-of-training-to-win-a-competition/solution.hpp
@@ -0,0 +1,30 @@
+#pragma once
+
+#include <numeric>
+#include <vector>
+
+class Solution {
+public:
+  static int minNumberOfHours(int initialEnergy, int initialExperience,
+                              const std::vector<int> &energy,
+                              const std::vector<int> &experience) {
+    return trainEnergy(initialEnergy, energy) +
+           trainExperience(initialExperience, experience);
+  }
+
+private:
+  static int trainEnergy(int initialEnergy, const std::vector<int> &energy) {
+    const auto sum = std::accumulate(energy.begin(), energy.end(), 0);
+    return std::max(0, sum - initialEnergy + 1);
+  }
+
+  static int trainExperience(int initialExperience,
+                             const std::vector<int> &experience) {
+    int max = 0, prev = 0;
+    for (auto e : experience) {
+      max = std::max(max, e - prev);
+      prev += e;
+    }
+    return std::max(0, max - initialExperience + 1);
+  }
+};
diff --git a/solutions/minimum-hours-of-training-to-win-a-competition/test.cpp b/solutions/minimum-hours-of-training-to-win-a-competition/test.cpp
new file mode 100644
index 00000000..0ab78642
--- /dev/null
+++ b/solutions/minimum-hours-of-training-to-win-a-competition/test.cpp
@@ -0,0 +1,37 @@
+#include <catch.hpp>
+
+#include <solution.hpp>
+
+TEST_CASE("Simple") {
+  struct TestCase {
+    int initialEnergy;
+    int initialExperience;
+    std::vector<int> energy;
+    std::vector<int> experience;
+    int expected;
+  };
+
+  std::vector<TestCase> test_cases{
+      {
+          .initialEnergy = 5,
+          .initialExperience = 3,
+          .energy{1, 4, 3, 2},
+          .experience{2, 6, 3, 1},
+          .expected = 8,
+      },
+      {
+          .initialEnergy = 2,
+          .initialExperience = 4,
+          .energy{1},
+          .experience{3},
+          .expected = 0,
+      },
+  };
+
+  for (const auto &[initialEnergy, initialExperience, energy, experience,
+                    expected] : test_cases) {
+    const auto actual = Solution::minNumberOfHours(
+        initialEnergy, initialExperience, energy, experience);
+    REQUIRE(expected == actual);
+  }
+}