1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
package main
import (
"fmt"
)
func minimumAverageDifference(nums []int) int {
ave_l := 0
num_count := len(nums)
if num_count == 1 {
return 0
}
ave_r := func(nums []int) int {
sum := 0
for i := 1; i < num_count; i++ {
sum += nums[i]
}
return sum
}(nums)
ave_l = nums[0]
// averageDifferences := make(map[int]int, num_count)
averageDifferences := make([]int, num_count, num_count)
averageDifferences[0] = ave_l - (ave_r / (num_count - 1))
if averageDifferences[0] < 0 {
averageDifferences[0] = -averageDifferences[0]
}
for i := 1; i < num_count-1; i++ {
ave_l = (ave_l + nums[i])
ave_r = (ave_r - nums[i])
averageDifferences[i] = (ave_l / (i + 1)) - (ave_r / (num_count - i - 1))
if averageDifferences[i] < 0 {
averageDifferences[i] = -averageDifferences[i]
}
// fmt.Println(ave_l, ave_r, averageDifferences[i])
}
ave_l = ave_l + nums[num_count-1]
averageDifferences[num_count-1] = ave_l / num_count
if averageDifferences[num_count-1] < 0 {
averageDifferences[num_count-1] = -averageDifferences[num_count-1]
}
min := averageDifferences[0]
min_k := 0
for k, v := range averageDifferences {
if v < min {
min = v
min_k = k
}
}
// fmt.Println(averageDifferences)
return min_k
}
func main() {
fmt.Println(minimumAverageDifference([]int{2, 5, 3, 9, 5, 3}))
fmt.Println(minimumAverageDifference([]int{0}))
fmt.Println(minimumAverageDifference([]int{1, 2, 3, 4, 5}))
fmt.Println(minimumAverageDifference([]int{5, 4, 3, 2, 1}))
fmt.Println(minimumAverageDifference(GetBig()))
fmt.Println(minimumAverageDifference([]int{2, 5, 3, 9, 5, 3}))
}
|
