Merge 4e57e22 into a09bb00

README.md

@@ -413,4 +413,5 @@ problems from
 * [Day 414](https://github.com/vaskoz/dailycodingproblem-go/issues/836)
 * [Day 415](https://github.com/vaskoz/dailycodingproblem-go/issues/838)
 * [Day 416](https://github.com/vaskoz/dailycodingproblem-go/issues/840)
+* [Day 417](https://github.com/vaskoz/dailycodingproblem-go/issues/842)
 

day417/problem.go

@@ -0,0 +1,35 @@
+package day417
+
+// LL is a linked list of integers.
+type LL struct {
+	Value int
+	Next  *LL
+}
+
+// RemoveConsecutiveSumZero removes all sublists that sum to zero.
+// Runs in O(N^2) time and O(1) space.
+func RemoveConsecutiveSumZero(head *LL) *LL {
+	fakeHead := &LL{0, head}
+
+	for start := fakeHead; start != nil; start = start.Next {
+		var sum int
+
+		var farthest *LL
+
+		var found bool
+
+		for next := start.Next; next != nil; next = next.Next {
+			sum += next.Value
+			if sum == 0 {
+				farthest = next.Next
+				found = true
+			}
+		}
+
+		if found {
+			start.Next = farthest
+		}
+	}
+
+	return fakeHead.Next
+}

day417/problem_test.go

@@ -0,0 +1,60 @@
+package day417
+
+import "testing"
+
+// nolint
+var testcases = []struct {
+	input, expected *LL
+}{
+	{
+		&LL{3, &LL{4, &LL{-7, &LL{5, &LL{-6, &LL{6, nil}}}}}},
+		&LL{5, nil},
+	},
+	{
+		&LL{3, &LL{4, &LL{5, &LL{-12, nil}}}},
+		nil,
+	},
+	{
+		&LL{3, &LL{4, &LL{-7, &LL{5, &LL{-5, &LL{6, nil}}}}}},
+		&LL{6, nil},
+	},
+	{
+		&LL{3, &LL{4, &LL{-7, &LL{3, &LL{5, &LL{-6, &LL{6, &LL{-5, nil}}}}}}}},
+		&LL{3, nil},
+	},
+	{
+		&LL{1, &LL{1, &LL{2, &LL{3, &LL{4, &LL{5, &LL{6, &LL{-5, nil}}}}}}}},
+		&LL{1, &LL{1, &LL{2, &LL{3, &LL{4, &LL{5, &LL{6, &LL{-5, nil}}}}}}}},
+	},
+}
+
+func equal(a, b *LL) bool {
+	for a != nil && b != nil {
+		if a.Value != b.Value {
+			return false
+		}
+
+		a = a.Next
+		b = b.Next
+	}
+
+	return a == nil && b == nil
+}
+
+func TestRemoveConsecutiveSumZero(t *testing.T) {
+	t.Parallel()
+
+	for tcid, tc := range testcases {
+		if result := RemoveConsecutiveSumZero(tc.input); !equal(result, tc.expected) {
+			t.Errorf("TCID%d doesn't match", tcid)
+		}
+	}
+}
+
+func BenchmarkRemoveConsecutiveSumZero(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		for _, tc := range testcases {
+			RemoveConsecutiveSumZero(tc.input)
+		}
+	}
+}