Merge pull request 6boris#155 from kylesliu/develop

Kyle Liu · web-flow · commit c57e5085081b · 2020-09-28T20:56:38.000+08:00
add 309 solution
diff --git a/src/0309.Best-Time-to-Buy-and-Sell-Stock-with-Cooldown/Solution.go b/src/0309.Best-Time-to-Buy-and-Sell-Stock-with-Cooldown/Solution.go
@@ -1,5 +1,21 @@
 package Solution
 
-func Solution(x bool) bool {
-	return x
+func maxProfit(prices []int) int {
+	sold, held, reset := -1<<31, -1<<31, 0
+
+	for _, price := range prices {
+		preSold := sold
+		sold = held + price
+		held = max(held, reset-price)
+		reset = max(reset, preSold)
+
+	}
+	return max(sold, reset)
+}
+
+func max(x, y int) int {
+	if x > y {
+		return x
+	}
+	return y
 }
diff --git a/src/0309.Best-Time-to-Buy-and-Sell-Stock-with-Cooldown/Solution_test.go b/src/0309.Best-Time-to-Buy-and-Sell-Stock-with-Cooldown/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", []int{1, 2, 3, 0, 2}, 3},
+		{"TestCase", []int{1}, 0},
 	}
 
 	//	开始测试
 	for i, c := range cases {
 		t.Run(c.name+" "+strconv.Itoa(i), func(t *testing.T) {
-			got := Solution(c.inputs)
+			got := maxProfit(c.inputs)
 			if !reflect.DeepEqual(got, c.expect) {
 				t.Fatalf("expected: %v, but got: %v, with inputs: %v",
 					c.expect, got, c.inputs)
