Solve '4. Median of Two Sorted Arrays'

Author: obalunenko

4-median-of-two-sorted-arrays/median.go

@@ -0,0 +1,30 @@
+package median
+
+import (
+	"fmt"
+	"sort"
+)
+
+func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
+	merged := make([]int, len(nums1)+len(nums2))
+	merged = append(nums1, nums2...)
+
+	sort.Ints(merged)
+
+	fmt.Printf("merged: %+v \n", merged)
+
+	var median float64
+	isbalanced := len(merged) % 2
+	if isbalanced == 0 {
+		middle := len(merged) / 2.0
+		fmt.Printf("middle [%v]\n", middle)
+
+		median = float64(merged[middle]+merged[middle-1]) / 2
+	} else {
+		middle := (len(merged) - 1) / 2
+		fmt.Printf("middle [%v]\n", middle)
+		median = float64(merged[(middle)])
+	}
+	fmt.Printf("median: %v \n", median)
+	return median
+}

4-median-of-two-sorted-arrays/median_test.go

@@ -0,0 +1,51 @@
+package median
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func Test_findMedianSortedArrays(t *testing.T) {
+	type args struct {
+		nums1 []int
+		nums2 []int
+	}
+	tests := []struct {
+		name string
+		args args
+		want float64
+	}{
+		{
+			name: "",
+			args: args{
+				nums1: []int{1, 3},
+				nums2: []int{2},
+			},
+			want: 2.0,
+		},
+		{
+			name: "",
+			args: args{
+				nums1: []int{1, 2},
+				nums2: []int{3, 4},
+			},
+			want: 2.5,
+		},
+		{
+			name: "",
+			args: args{
+				nums1: []int{1, 2, 5, 9},
+				nums2: []int{0, 4, 10, 11},
+			},
+			want: 4.5,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := findMedianSortedArrays(tt.args.nums1, tt.args.nums2)
+			assert.Equal(t, tt.want, got)
+
+		})
+	}
+}

4-median-of-two-sorted-arrays/spec.md

@@ -0,0 +1,24 @@
+# 4. Median of Two Sorted Arrays
+
+<https://leetcode.com/problems/median-of-two-sorted-arrays/>
+
+There are two sorted arrays nums1 and nums2 of size m and n respectively.
+
+Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
+
+You may assume nums1 and nums2 cannot be both empty.
+
+Example 1:
+
+```text
+nums1 = [1, 3]
+nums2 = [2]
+
+The median is 2.0
+Example 2:
+
+nums1 = [1, 2]
+nums2 = [3, 4]
+
+The median is (2 + 3)/2 = 2.5
+```