Необходимо вернуть слайс целых чисел каждый элемент которого содержит произведения всех элементов входного слайса кроме элемента с совпадающим индексом.
Ограничения: Не использовать деление, сложность O(n),
| Go | 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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
| // [url]https://leetcode.com/studyplan/top-interview-150/[/url]
package topInterview
// productExceptSelf
//
// 238. Product of Array Except Self
// Medium
// Topics
// Companies
// Hint
// Given an integer array nums, return an array answer such that answer[i] is equal to the product of all the elements of nums except nums[i].
//
// The product of any prefix or suffix of nums is guaranteed to fit in a 32-bit integer.
//
// You must write an algorithm that runs in O(n) time and without using the division operation.
//
// Example 1:
//
// Input: nums = [1,2,3,4]
// Output: [24,12,8,6]
// Example 2:
//
// Input: nums = [-1,1,0,-3,3]
// Output: [0,0,9,0,0]
//
// Constraints:
//
// 2 <= nums.length <= 105
// -30 <= nums[i] <= 30
// The input is generated such that answer[i] is guaranteed to fit in a 32-bit integer.
func productExceptSelf(nums []int) []int {
// Инициировать переменную содержащую накопленное произведение единицей
cumulativeProduct := 1
// Инициировать переменную содержащую индекс итерируемого элемента слайса нулем
index := 0
// Инициировать промежуточный слайс нулями, длиной аналогичной nums
output := make([]int, len(nums))
// Запустить цикл
for {
// Значение элемента промежуточного слайса с индексом Index
// прировнять к значению переменной накопленной произведения
output[index] = cumulativeProduct
// Значение накопленного произведения приравнять к произведению накопленного
// произведения и значения элемента nums c индексом Index
cumulativeProduct *= nums[index]
// Если значение Index выходит за приделы крайнего элемента слайса nums
if index >= len(nums)-1 {
// Прервать цикл
break
}
// Инкрементировать Index
index++
}
// Значение накопленного предвидения прировнять к единице
cumulativeProduct = 1
for {
// Значение элемента промежуточного слайса с индексом Index прировнять к произведению
// значение элемента промежуточного слайса с индексом Index и накопленного произведения
output[index] *= cumulativeProduct
// Значение накопленного произведения прировнять к произведению наколенного произведения
// и элемента слайса nums с индексом Index
cumulativeProduct *= nums[index]
// Если Index меньше нуля
if index <= 0 {
// Прервать цикл
break
}
// Декриментировать Index
index--
}
// Вернуть промежуточный слайс
return output
} |
|
| Go | 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
| func TestProductExceptSelf(t *testing.T) {
data := []struct {
nums []int
expected []int
}{
{
[]int{1, 2, 3, 4},
[]int{24, 12, 8, 6},
},
{
[]int{-1, 1, 0, -3, 3},
[]int{0, 0, 9, 0, 0},
},
}
for i, datum := range data {
result := productExceptSelf(datum.nums)
if !reflect.DeepEqual(result, datum.expected) {
t.Errorf("unexpected result for test index %d expected [%+v] got [%+v]", i, datum.expected, result)
}
}
} |
|
| Code | 1
2
3
| === RUN TestProductExceptSelf
--- PASS: TestProductExceptSelf (0.00s)
PASS |
|
| https://github.com/alhaos/problems
|