✨ Added solution to 198

Sathish Babu · Sathish Babu · commit b85896288084 · 2020-06-06T13:42:39.000+05:30
diff --git a/src/0198.House-Robber/Solution.go b/src/0198.House-Robber/Solution.go
@@ -36,6 +36,22 @@ func rob3(nums []int) int {
 	return max(robEven, robOdd)
 }
 
+func rob4(nums []int) int {
+	n := len(nums)
+	if n == 0 {
+		return 0
+	}
+	if n == 1 {
+		return nums[0]
+	}
+	dp := make([]int, n)
+	dp[0], dp[1] = nums[0], max(nums[0], nums[1])
+	for i := 2; i < n; i++ {
+		dp[i] = max(nums[i]+dp[i-2], dp[i-1])
+	}
+	return dp[n-1]
+}
+
 func max(x, y int) int {
 	if x > y {
 		return x
diff --git a/src/0198.House-Robber/Solution_test.go b/src/0198.House-Robber/Solution_test.go
@@ -2,6 +2,7 @@ package Solution
 
 import (
 	"reflect"
+	"strconv"
 	"testing"
 )
 
@@ -75,6 +76,32 @@ func TestSolution3(t *testing.T) {
 	}
 }
 
+func TestSolution4(t *testing.T) {
+	//	测试用例
+	cases := []struct {
+		name   string
+		inputs []int
+		expect int
+	}{
+		{"TestCacse", []int{}, 0},
+		{"TestCacse", []int{3}, 3},
+		{"TestCacse", []int{1, 2, 3, 1}, 4},
+		{"TestCacse", []int{2, 7, 9, 3, 1}, 12},
+		{"TestCacse", []int{5, 2, 6, 7, 3, 1}, 14},
+	}
+
+	//	开始测试
+	for i, c := range cases {
+		t.Run(c.name+" "+strconv.Itoa(i), func(t *testing.T) {
+			got := rob4(c.inputs)
+			if !reflect.DeepEqual(got, c.expect) {
+				t.Fatalf("expected: %v, but got: %v, with inputs: %v",
+					c.expect, got, c.inputs)
+			}
+		})
+	}
+}
+
 //	压力测试
 func BenchmarkSolution(b *testing.B) {
 
