uncommited problems

candy/sol.go

@@ -0,0 +1,27 @@
+package main
+
+func candy(ratings []int) int {
+	count := 1
+	for idx := 0; idx < len(ratings)-1; {
+		for idx < len(ratings)-1 && ratings[idx+1] == ratings[idx] {
+			count++
+			idx++
+		}
+		next_inc := 1
+		for idx < len(ratings)-1 && ratings[idx+1] > ratings[idx] {
+			next_inc++
+			idx++
+			count += next_inc
+		}
+		next_dec := 1
+		for idx < len(ratings)-1 && ratings[idx+1] < ratings[idx] {
+			count += next_dec
+			next_dec++
+			idx++
+		}
+		if next_dec > next_inc {
+			count += next_dec - next_inc
+		}
+	}
+	return count
+}