feature DivingBoard

andycom · andycom · commit 4de23ad32696 · 2023-03-18T17:44:34.000+08:00
diff --git a/src/main/java/com/fancv/leetCode/arrays/DivingBoard.java b/src/main/java/com/fancv/leetCode/arrays/DivingBoard.java
@@ -0,0 +1,27 @@
+package com.fancv.leetCode.arrays;
+
+public class DivingBoard {
+
+    public static void main(String[] args) {
+
+        System.out.println(divingBoard(1, 1, 123));
+    }
+
+    public static int[] divingBoard(int shorter, int longer, int k) {
+
+
+        if (k == 0) {
+            return new int[k];
+        }
+        if (shorter == longer) {
+            return new int[]{shorter * k};
+        }
+        int[] r = new int[k + 1];
+        for (int i = 0; i < k + 1; i++) {
+
+            r[i] = shorter * (k - i) + longer * i;
+        }
+        return r;
+
+    }
+}
