add 441 solution

6boris · 6boris · commit 9c3b9445f446 · 2020-07-06T19:47:00.000+08:00
diff --git a/src/0441.Arranging-Coins/Solution.go b/src/0441.Arranging-Coins/Solution.go
@@ -1,5 +1,12 @@
 package Solution
 
-func Solution(x bool) bool {
-	return x
+import "math"
+
+func arrangeCoins(n int) int {
+	if n <= 0 {
+		return 0
+	}
+
+	x := math.Sqrt(2*float64(n)+0.25) - 0.5
+	return int(x)
 }
diff --git a/src/0441.Arranging-Coins/Solution_test.go b/src/0441.Arranging-Coins/Solution_test.go
@@ -10,18 +10,17 @@ func TestSolution(t *testing.T) {
 	//	测试用例
 	cases := []struct {
 		name   string
-		inputs bool
-		expect bool
+		inputs int
+		expect int
 	}{
-		{"TestCase", true, true},
-		{"TestCase", true, true},
-		{"TestCase", false, false},
+		{"TestCase", 5, 2},
+		{"TestCase", 8, 3},
 	}
 
 	//	开始测试
 	for i, c := range cases {
 		t.Run(c.name+" "+strconv.Itoa(i), func(t *testing.T) {
-			got := Solution(c.inputs)
+			got := arrangeCoins(c.inputs)
 			if !reflect.DeepEqual(got, c.expect) {
 				t.Fatalf("expected: %v, but got: %v, with inputs: %v",
 					c.expect, got, c.inputs)
