2015-day01

This commit is contained in:
alexchao26
2020-12-26 00:54:58 -05:00
parent 94d2e879b0
commit 1d09030c28
2 changed files with 61 additions and 0 deletions
+35
View File
@@ -0,0 +1,35 @@
package main
import (
"flag"
"fmt"
"github.com/alexchao26/advent-of-code-go/util"
)
func main() {
var part int
flag.IntVar(&part, "part", 1, "part 1 or 2")
flag.Parse()
fmt.Println("Running part", part)
ans := notQuiteLisp(util.ReadFile("./input.txt"), part)
fmt.Println("Output:", ans)
}
func notQuiteLisp(input string, part int) int {
var level int
for i, r := range input {
if r == '(' {
level++
} else {
level--
}
if part == 2 && level == -1 {
return i + 1
}
}
return level
}
+26
View File
@@ -0,0 +1,26 @@
package main
import (
"testing"
"github.com/alexchao26/advent-of-code-go/util"
)
func Test_notQuiteLisp(t *testing.T) {
tests := []struct {
name string
input string
part int
want int
}{
{"actual", util.ReadFile("input.txt"), 1, 232},
{"actual", util.ReadFile("input.txt"), 2, 1783},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if got := notQuiteLisp(tt.input, tt.part); got != tt.want {
t.Errorf("notQuiteLisp() = %v, want %v", got, tt.want)
}
})
}
}