Merge 51fd354 into e56d2e1

README.md

@@ -245,3 +245,4 @@ problems from
 * [Day 234](https://github.com/vaskoz/dailycodingproblem-go/issues/481)
 * [Day 235](https://github.com/vaskoz/dailycodingproblem-go/issues/483)
 * [Day 236](https://github.com/vaskoz/dailycodingproblem-go/issues/485)
+* [Day 237](https://github.com/vaskoz/dailycodingproblem-go/issues/487)

day237/problem.go

@@ -0,0 +1,56 @@
+// Package day237 has a note.
+// NOTE: I didn't provide an API for a k-ary
+// tree. It's up to the user to build a valid
+// k-ary tree.
+package day237
+
+// KaryTree is a k-ary tree of integers.
+type KaryTree struct {
+	Value    int
+	Children []*KaryTree
+}
+
+// IsTreeSymmetric answers if its data and shape remain
+// unchanged when it is reflected about the root node.
+func IsTreeSymmetric(tree *KaryTree) bool {
+	if tree == nil {
+		return true
+	}
+	level := tree.Children
+	for {
+		nextLevel := make([]*KaryTree, 0, len(level)*2)
+		for _, l := range level {
+			if l == nil {
+				nextLevel = append(nextLevel, l)
+			} else {
+				nextLevel = append(nextLevel, l.Children...)
+			}
+		}
+		for i := 0; i < len(nextLevel)/2; i++ {
+			if !equal(nextLevel[i], nextLevel[len(nextLevel)-1-i]) {
+				return false
+			}
+		}
+		done := true
+		for i := range nextLevel {
+			if nextLevel[i] != nil {
+				done = false
+				break
+			}
+		}
+		if done {
+			break
+		}
+		level = nextLevel
+	}
+	return true
+}
+
+func equal(a, b *KaryTree) bool {
+	if a == nil && b == nil {
+		return true
+	} else if a != nil && b != nil && a.Value == b.Value {
+		return true
+	}
+	return false
+}

day237/problem_test.go

@@ -0,0 +1,61 @@
+package day237
+
+import "testing"
+
+var testcases = []struct {
+	root      *KaryTree
+	symmetric bool
+}{
+	{&KaryTree{
+		4,
+		[]*KaryTree{
+			{
+				3,
+				[]*KaryTree{{9, nil}, nil, nil},
+			},
+			{
+				5,
+				[]*KaryTree{nil, nil, nil},
+			},
+			{
+				3,
+				[]*KaryTree{nil, nil, {9, nil}},
+			},
+		},
+	}, true},
+	{nil, true},
+	{&KaryTree{
+		4,
+		[]*KaryTree{
+			{
+				3,
+				[]*KaryTree{nil, {9, nil}, nil},
+			},
+			{
+				5,
+				[]*KaryTree{nil, nil, nil},
+			},
+			{
+				3,
+				[]*KaryTree{nil, nil, {9, nil}},
+			},
+		},
+	}, false},
+}
+
+func TestIsTreeSymmetric(t *testing.T) {
+	t.Parallel()
+	for _, tc := range testcases {
+		if result := IsTreeSymmetric(tc.root); result != tc.symmetric {
+			t.Errorf("Expected %v, got %v", tc.symmetric, result)
+		}
+	}
+}
+
+func BenchmarkIsTreeSymmetric(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		for _, tc := range testcases {
+			IsTreeSymmetric(tc.root)
+		}
+	}
+}