2600. K Items With the Maximum Sum

donutloop · donutloop · commit c6682aedd15c · 2024-03-11T17:08:17.000+01:00
diff --git a/K Items With the Maximum Sum/kata.go b/K Items With the Maximum Sum/kata.go
@@ -0,0 +1,24 @@
+package kata
+
+func kItemsWithMaximumSum(numOnes int, numZeros int, numNegOnes int, k int) int {
+	var sum int
+	if numOnes >= k {
+		sum += 1 * k
+		return sum
+	} else if numOnes > 0 {
+		sum += numOnes * 1
+		k = k - numOnes
+	}
+	if numZeros >= k {
+		return sum
+	} else if numZeros > 0 {
+		k = k - numZeros
+	}
+	if numNegOnes >= k {
+		sum += -1 * k
+		return sum
+	} else if numNegOnes > 0 {
+		sum += -1 * numNegOnes
+	}
+	return sum
+}
