20201110_find_lost_number

This commit is contained in:
VicRen
2020-11-10 14:29:29 +08:00
parent 37ac2b7c2b
commit b7f6c8451b
2 changed files with 17 additions and 2 deletions

View File

@@ -24,8 +24,8 @@ func findLostNumber(nums []int) int {
for _, n := range nums { for _, n := range nums {
sum += n sum += n
} }
n := len(nums) l := len(nums)
return n*(n+1)/2 - sum return l*(l+1)/2 - sum
} }
func findLostNumber2(nums []int) int { func findLostNumber2(nums []int) int {

View File

@@ -23,6 +23,16 @@ func Test_findLostNumber(t *testing.T) {
args{[]int{9, 6, 4, 2, 3, 5, 7, 0, 1}}, args{[]int{9, 6, 4, 2, 3, 5, 7, 0, 1}},
8, 8,
}, },
{
"example_3",
args{[]int{8, 6, 4, 2, 3, 5, 7, 0, 1}},
9,
},
{
"example_4",
args{[]int{9, 8, 6, 4, 2, 3, 5, 7, 1}},
0,
},
} }
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
@@ -52,6 +62,11 @@ func Test_findLostNumber2(t *testing.T) {
args{[]int{9, 6, 4, 2, 3, 5, 7, 0, 1}}, args{[]int{9, 6, 4, 2, 3, 5, 7, 0, 1}},
8, 8,
}, },
{
"example_3",
args{[]int{9, 8, 6, 4, 2, 3, 5, 7, 1}},
0,
},
} }
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {