feat: type/slicesutil: add MatrixIntColSums(), SliceIntSum()

grokify · Mar 10, 2024 · b56f7f2 · b56f7f2
1 parent d495a00
commit b56f7f2
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/type/slicesutil/int.go b/type/slicesutil/int.go
@@ -0,0 +1,22 @@
+package slicesutil
+
+func SliceIntSum(s []int) int {
+	sum := 0
+	for _, si := range s {
+		sum += si
+	}
+	return sum
+}
+
+func MatrixIntColSums(m [][]int) []int {
+	sums := []int{}
+	for _, r := range m {
+		for ci, c := range r {
+			for ci >= len(sums) {
+				sums = append(sums, 0)
+			}
+			sums[ci] += c
+		}
+	}
+	return sums
+}