1089 wrong answer

Author: aQuaYi

Algorithms/1089.duplicate-zeros/duplicate-zeros.go

@@ -1,4 +1,27 @@
 package problem1089
 
-func duplicateZeros(arr []int) {
+func duplicateZeros(A []int) {
+	n := len(A)
+	//
+	i, count := 0, 0
+	for i+count < n {
+		if A[i] == 0 {
+			count++
+		}
+		i++
+	}
+	i--
+	//
+	j := n - 1
+	for count > 0 {
+		if A[i] == 0 {
+			A[j-1], A[j] = 0, 0
+			j -= 2
+			count--
+		} else {
+			A[j] = A[i]
+			j--
+		}
+		i--
+	}
 }

Algorithms/1089.duplicate-zeros/duplicate-zeros_test.go

@@ -12,6 +12,11 @@ var tcs = []struct {
 	ans []int
 }{
 
+	{
+		[]int{0, 0, 0, 0, 0, 0, 0},
+		[]int{0, 0, 0, 0, 0, 0, 0},
+	},
+
 	{
 		[]int{1, 0, 2, 3, 0, 4, 5, 0},
 		[]int{1, 0, 0, 2, 3, 0, 0, 4},