Improve day 5

day05/ex2/main.go

@@ -11,12 +11,13 @@ func main() {
   stacks, movements := common.Parse(*bufio.NewScanner(os.Stdin))
 
   for _, movement := range(movements) {
-    top := []byte{}
+    topStack := common.Stack{}
     for i := 0; i < movement.GetN(); i++ {
-      top = append(top, stacks[movement.GetStart() - 1].Pop())
+      top, _ := stacks[movement.GetStart() - 1].Pop()
+      topStack.Push(top)
     }
-    for i := len(top) - 1; i >= 0; i-- {
-      stacks[movement.GetEnd() - 1].Push(top[i])
+    for top, left := topStack.Pop(); left; top, left = topStack.Pop() {
+      stacks[movement.GetEnd() - 1].Push(top)
     }
   }