https://leetcode.cn/contest/cnunionpay2022/problems/wMGN0t/

Author: masx200

README.md

@@ -45,6 +45,8 @@ Step 2. Add the dependency
 
 <summary>展开查看</summary>
 
+https://leetcode.cn/contest/cnunionpay2022/problems/wMGN0t/
+
 https://leetcode.cn/contest/cnunionpay2022/problems/6olJmJ/
 
 https://leetcode.cn/problems/number-of-great-partitions

wMGN0t/index.ts

@@ -0,0 +1,19 @@
+function storedEnergy(
+    storeLimit: number,
+    power: number[],
+    supply: number[][],
+): number {
+    const { min, max } = Math;
+    let sum = 0;
+    let time = 0;
+    for (let i = 0; i < power.length; i++) {
+        while (time + 1 < supply.length && supply[time + 1][0] <= i) time++;
+        if (power[i] > supply[time][2]) {
+            sum = min(sum + power[i] - supply[time][2], storeLimit);
+        } else if (power[i] < supply[time][1]) {
+            sum = max(sum - supply[time][1] + power[i], 0);
+        }
+    }
+    return sum;
+}
+export default storedEnergy;

wMGN0t/test.ts

@@ -0,0 +1,16 @@
+import { assertEquals } from "asserts";
+import storedEnergy from "./index.ts";
+Deno.test("storedEnergy", () => {
+    const storeLimit = 10,
+        power = [1, 3, 4, 3, 6],
+        supply = [[0, 2, 3]],
+        输出 = 4;
+    assertEquals(storedEnergy(storeLimit, power, supply), 输出);
+});
+Deno.test("storedEnergy", () => {
+    const storeLimit = 6,
+        power = [6, 5, 2, 1, 0],
+        supply = [[0, 1, 2], [2, 3, 3]],
+        输出 = 0;
+    assertEquals(storedEnergy(storeLimit, power, supply), 输出);
+});