June 1: Distribute candies I [E, M]

Math problem in senior high
aucker · Jun 1, 2024 · a67ca17 · a67ca17
1 parent 2ed15bf
commit a67ca17
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/daily/June1.cpp b/daily/June1.cpp
@@ -0,0 +1,17 @@
+#include <iostream>
+using namespace std;
+
+class Solution {
+  int c2(int n) { return n > 1 ? n * (n - 1) / 2 : 0; }
+
+ public:
+  /**
+   * LC: 2928: Distribute Candies among children I
+   * MATH
+   * Time/Space: O(1)
+   */
+  int distributeCandies(int n, int limit) {
+    return c2(n + 2) - 3 * c2(n - limit + 1) + 3 * c2(n - 2 * limit) -
+           c2(n - 3 * limit - 1);
+  }
+};