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 0ebfff8

Browse files
author
Sandy
authored
Merge pull request #84 from openset/develop
Add: median_of_two_sorted_arrays
2 parents 555f734 + 3567731 commit 0ebfff8

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 によって変換されたページ (->オリジナル) /