Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly
Sign up
Appearance settings

Commit 3567731

Browse files
author
Openset
committed
Add: median_of_two_sorted_arrays
1 parent c93fd34 commit 3567731

File tree

2 files changed

+110
-0
lines changed

2 files changed

+110
-0
lines changed
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,31 @@
11
package median_of_two_sorted_arrays
2+
3+
func findMedianSortedArrays(nums1 []int, nums2 []int) float64 {
4+
m, n := len(nums1), len(nums2)
5+
s := make([]int, 0, m+n)
6+
i, j := 0, 0
7+
for i < m && j < n {
8+
v1 := nums1[i]
9+
v2 := nums2[j]
10+
if v1 <= v2 {
11+
s = append(s, v1)
12+
i++
13+
} else {
14+
s = append(s, v2)
15+
j++
16+
}
17+
}
18+
if i < m {
19+
s = append(s, nums1[i:]...)
20+
} else {
21+
s = append(s, nums2[j:]...)
22+
}
23+
count := m + n
24+
if count&1 == 1 {
25+
k := (count - 1) / 2
26+
return float64(s[k])
27+
} else {
28+
k := count / 2
29+
return (float64(s[k-1]) + float64(s[k])) / 2
30+
}
31+
}
Lines changed: 80 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,81 @@
11
package median_of_two_sorted_arrays
2+
3+
import "testing"
4+
5+
type caseType struct {
6+
num1 []int
7+
num2 []int
8+
expected float64
9+
}
10+
11+
func TestFindMedianSortedArrays(t *testing.T) {
12+
tests := [...]caseType{
13+
{
14+
num1: []int{1, 3},
15+
num2: []int{2},
16+
expected: 2.0,
17+
},
18+
{
19+
num1: []int{1, 2},
20+
num2: []int{3, 4},
21+
expected: 2.5,
22+
},
23+
{
24+
num1: []int{1, 3, 5, 7},
25+
num2: []int{2, 4},
26+
expected: 3.5,
27+
},
28+
{
29+
num1: []int{1, 3, 5},
30+
num2: []int{},
31+
expected: 3,
32+
},
33+
{
34+
num1: []int{1, 2},
35+
num2: []int{3},
36+
expected: 2,
37+
},
38+
{
39+
num1: []int{1, 1},
40+
num2: []int{1, 1},
41+
expected: 1,
42+
},
43+
{
44+
num1: []int{1, 1, 1},
45+
num2: []int{1, 1},
46+
expected: 1,
47+
},
48+
{
49+
num1: []int{1, 1},
50+
num2: []int{1, 1, 1},
51+
expected: 1,
52+
},
53+
{
54+
num1: []int{1},
55+
num2: []int{2, 3, 4},
56+
expected: 2.5,
57+
},
58+
{
59+
num1: []int{2, 3, 4},
60+
num2: []int{1},
61+
expected: 2.5,
62+
},
63+
{
64+
num1: []int{},
65+
num2: []int{1},
66+
expected: 1,
67+
},
68+
{
69+
num1: []int{},
70+
num2: []int{1, 1},
71+
expected: 1,
72+
},
73+
}
74+
75+
for _, tc := range tests {
76+
output := findMedianSortedArrays(tc.num1, tc.num2)
77+
if output != tc.expected {
78+
t.Fatalf("input: %v %v, output: %v, expected: %v", tc.num1, tc.num2, output, tc.expected)
79+
}
80+
}
81+
}

0 commit comments

Comments
(0)

AltStyle によって変換されたページ (->オリジナル) /