aboutsummaryrefslogtreecommitdiff
path: root/challenge-333/deadmarshal/go/ch2.go
blob: a93532127b0aed50203f1b3782110bcddf3ab39f (plain)
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
package main

import (
	"fmt"
)

func duplicateZeros(arr []int) {
	n := len(arr)
	i, k := -1, 0
	for k < n {
		i, k = i+1, k+1
		if arr[i] == 0 {
			k++
		}
	}
	j := n - 1
	if k == n+1 {
		arr[j] = 0
		i, j = i-1, j-1
	}
	for j >= 0 {
		arr[j] = arr[i]
		if arr[i] == 0 {
			j--
			arr[j] = arr[i]
		}
		i, j = i-1, j-1
	}
}

func main() {
	a1 := []int{1, 0, 2, 3, 0, 4, 5, 0}
	a2 := []int{1, 2, 3}
	a3 := []int{1, 2, 3, 0}
	a4 := []int{0, 0, 1, 2}
	a5 := []int{1, 2, 0, 3, 4}
	duplicateZeros(a1)
	duplicateZeros(a2)
	duplicateZeros(a3)
	duplicateZeros(a4)
	duplicateZeros(a5)
	fmt.Println(a1)
	fmt.Println(a2)
	fmt.Println(a3)
	fmt.Println(a4)
	fmt.Println(a5)
}