2570. Merge Two 2D Arrays by Summing Values

donutloop · donutloop · commit b98b147953b5 · 2024-03-11T14:55:03.000+01:00
diff --git a/Merge Two 2D Arrays by Summing Values/kata.go b/Merge Two 2D Arrays by Summing Values/kata.go
@@ -0,0 +1,31 @@
+package kata
+
+func mergeArrays(nums1 [][]int, nums2 [][]int) [][]int {
+	merged := make([][]int, 0)
+
+	i, j := 0, 0
+	for i < len(nums1) || j < len(nums2) {
+		if i < len(nums1) && j < len(nums2) {
+			if nums1[i][0] == nums2[j][0] {
+				nums1[i][1] += nums2[j][1]
+				merged = append(merged, nums1[i])
+				i++
+				j++
+			} else if nums1[i][0] < nums2[j][0] {
+				merged = append(merged, nums1[i])
+				i++
+			} else {
+				merged = append(merged, nums2[j])
+				j++
+			}
+		} else if i < len(nums1) {
+			merged = append(merged, nums1[i])
+			i++
+		} else if j < len(nums2) {
+			merged = append(merged, nums2[j])
+			j++
+		}
+	}
+
+	return merged
+}
