66. Plus One

Author: vfedorovatlas

plus-one/Readme.md

@@ -0,0 +1,3 @@
+# 66. Plus One
+
+Source: https://leetcode.com/problems/plus-one/description/

plus-one/main.go

@@ -0,0 +1,17 @@
+package leecode
+
+func plusOne(digits []int) []int {
+	plusOne := true
+	// start from the last element
+	for i := len(digits) - 1; i >= 0; i-- {
+		if digits[i] < 9 {
+			digits[i]++
+			return digits
+		}
+		digits[i] = 0
+	}
+	if plusOne {
+		digits = append([]int{1}, digits...)
+	}
+	return digits
+}

plus-one/main_test.go

@@ -0,0 +1,27 @@
+package leecode
+
+import (
+	"fmt"
+	"reflect"
+	"testing"
+)
+
+func TestDominantIndex(t *testing.T) {
+	testCases := []struct {
+		Input  []int
+		Output []int
+	}{
+		{
+			Input:  []int{9},
+			Output: []int{1, 0},
+		},
+	}
+	for _, tc := range testCases {
+		t.Run(fmt.Sprintf("%v:%d", tc.Input, tc.Output), func(t *testing.T) {
+			actual := plusOne(tc.Input)
+			if !reflect.DeepEqual(actual, tc.Output) {
+				t.Errorf("expected %d, but get %d", tc.Output, actual)
+			}
+		})
+	}
+}