122 finish

Author: aQua

Algorithms/0122.best-time-to-buy-and-sell-stock-ii/best-time-to-buy-and-sell-stock-ii.go

@@ -1,5 +1,13 @@
 package Problem0122
 
 func maxProfit(prices []int) int {
+	res := 0
 
+	for i := 1; i < len(prices); i++ {
+		if prices[i] > prices[i-1] {
+			res += prices[i] - prices[i-1]
+		}
+	}
+
+	return res
 }

Algorithms/0122.best-time-to-buy-and-sell-stock-ii/best-time-to-buy-and-sell-stock-ii_test.go

@@ -1,8 +1,8 @@
 package Problem0122
 
 import (
-	"testing"
 	"fmt"
+	"testing"
 
 	"github.com/stretchr/testify/assert"
 )
@@ -19,30 +19,47 @@ type para struct {
 
 // ans 是答案
 type ans struct {
-	one  int 
+	one int
 }
 
 func Test_Problem0122(t *testing.T) {
 	ast := assert.New(t)
 
 	qs := []question{
+		question{
+			para{
+				[]int{7, 2, 1, 5, 3, 6, 4},
+			},
+			ans{
+				7,
+			},
+		},
 
 		question{
 			para{
-					,
+				[]int{7, 1, 5, 3, 6, 4},
 			},
 			ans{
-					,
+				7,
 			},
 		},
-	
+
+		question{
+			para{
+				[]int{7, 6, 5, 4, 3, 2, 1},
+			},
+			ans{
+				0,
+			},
+		},
+
 		// 如需多个测试，可以复制上方元素。
 	}
 
 	for _, q := range qs {
 		a, p := q.ans, q.para
 		fmt.Printf("~~%v~~\n", p)
 
-		ast.Equal(a.one, maxProfit(p.prices  ), "输入:%v", p)
+		ast.Equal(a.one, maxProfit(p.prices), "输入:%v", p)
 	}
 }