20201111: sf

This commit is contained in:
VicRen
2020-11-11 09:04:36 +08:00
parent b7f6c8451b
commit 88658ff1d5
2 changed files with 199 additions and 0 deletions

65
20201111/main.go Normal file
View File

@@ -0,0 +1,65 @@
package main
import (
"errors"
"fmt"
"math/rand"
)
func main() {
r := rand.Intn(100)
fmt.Println("r:", r)
var src []int
for i := 0; i < 100; i++ {
src = append(src, i)
if i == r {
continue
}
src = append(src, i)
}
n, _ := findSingleNumber(src)
fmt.Println("lost number:", n)
n, _ = findSingleNumber2(src)
fmt.Println("lost number2:", n)
n, _ = findSingleNumber3(src)
fmt.Println("lost number3:", n)
}
func findSingleNumber(src []int) (int, error) {
m := map[int]int{}
for _, n := range src {
m[n] += 1
}
for k, v := range m {
if v == 1 {
return k, nil
}
}
return 0, errors.New("not found")
}
func findSingleNumber2(src []int) (int, error) {
m := map[int]struct{}{}
for _, n := range src {
if _, ok := m[n]; ok {
delete(m, n)
} else {
m[n] = struct{}{}
}
}
for k := range m {
return k, nil
}
return 0, errors.New("not found")
}
func findSingleNumber3(src []int) (int, error) {
num := 0
for _, n := range src {
num = num ^ n
}
if num == 0 {
return 0, errors.New("not found")
}
return num, nil
}

134
20201111/main_test.go Normal file
View File

@@ -0,0 +1,134 @@
package main
import (
"testing"
)
func Test_findSingleNumber(t *testing.T) {
type args struct {
src []int
}
tests := []struct {
name string
args args
want int
wantErr bool
}{
{
"example_1",
args{[]int{2, 2, 1}},
1,
false,
},
{
"example_2",
args{[]int{4, 1, 2, 1, 2}},
4,
false,
},
{
"example_3",
args{[]int{4, 4, 1, 2, 1, 2}},
0,
true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := findSingleNumber(tt.args.src)
if (err != nil) != tt.wantErr {
t.Errorf("findSingleNumber() error = %v, wantErr %v", err, tt.wantErr)
return
}
if got != tt.want {
t.Errorf("findSingleNumber() = %v, want %v", got, tt.want)
}
})
}
}
func Test_findSingleNumber2(t *testing.T) {
type args struct {
src []int
}
tests := []struct {
name string
args args
want int
wantErr bool
}{
{
"example_1",
args{[]int{2, 2, 1}},
1,
false,
},
{
"example_2",
args{[]int{4, 1, 2, 1, 2}},
4,
false,
},
{
"example_3",
args{[]int{4, 4, 1, 2, 1, 2}},
0,
true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := findSingleNumber2(tt.args.src)
if (err != nil) != tt.wantErr {
t.Errorf("findSingleNumber2() error = %v, wantErr %v", err, tt.wantErr)
return
}
if got != tt.want {
t.Errorf("findSingleNumber2() = %v, want %v", got, tt.want)
}
})
}
}
func Test_findSingleNumber3(t *testing.T) {
type args struct {
src []int
}
tests := []struct {
name string
args args
want int
wantErr bool
}{
{
"example_1",
args{[]int{2, 2, 1}},
1,
false,
},
{
"example_2",
args{[]int{4, 1, 2, 1, 2}},
4,
false,
},
{
"example_3",
args{[]int{4, 4, 1, 2, 1, 2}},
0,
true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := findSingleNumber3(tt.args.src)
if (err != nil) != tt.wantErr {
t.Errorf("findSingleNumber3() error = %v, wantErr %v", err, tt.wantErr)
return
}
if got != tt.want {
t.Errorf("findSingleNumber3() = %v, want %v", got, tt.want)
}
})
}
}