AOC 2024 Day 3 #23
|
@ -24,36 +24,18 @@ func (r result) String() string { return fmt.Sprintf("%#v", r) }
|
||||||
func run(read io.Reader) (*result, error) {
|
func run(read io.Reader) (*result, error) {
|
||||||
|
|
||||||
sum := 0
|
sum := 0
|
||||||
sum2 := 0
|
|
||||||
|
|
||||||
lexer := Lexer(read)
|
lexer := Lexer(read)
|
||||||
active := true
|
for typ, token := range lexer.Iter() {
|
||||||
|
|
||||||
for typ, _ := range lexer.Iter() {
|
if m := readMul(lexer); m != 0 {
|
||||||
// fmt.Println(typ, string(token))
|
fmt.Println(typ, string(token))
|
||||||
switch {
|
sum += m
|
||||||
case typ == "TokMUL":
|
|
||||||
if m := readMul(lexer); m != 0 {
|
|
||||||
sum += m
|
|
||||||
if active {
|
|
||||||
sum2 += m
|
|
||||||
}
|
|
||||||
}
|
|
||||||
case typ == "TokDONT":
|
|
||||||
if readDont(lexer) {
|
|
||||||
println("don't()", active)
|
|
||||||
active = false
|
|
||||||
}
|
|
||||||
|
|
||||||
case typ == "TokDO":
|
|
||||||
if readDo(lexer) {
|
|
||||||
println("do()", active)
|
|
||||||
active = true
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return &result{sum, sum2}, nil
|
|
||||||
|
return &result{sum, -1}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func Lexer(in io.Reader) *lexer {
|
func Lexer(in io.Reader) *lexer {
|
||||||
|
@ -63,8 +45,7 @@ func Lexer(in io.Reader) *lexer {
|
||||||
buf = buf[:s]
|
buf = buf[:s]
|
||||||
|
|
||||||
for len(buf) > 0 {
|
for len(buf) > 0 {
|
||||||
for i, r := range string(buf) {
|
for _, r := range string(buf) {
|
||||||
println(i, r)
|
|
||||||
if !yield(r) {
|
if !yield(r) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
@ -90,8 +71,8 @@ type lexer struct {
|
||||||
iter func() (rune, bool)
|
iter func() (rune, bool)
|
||||||
stop func()
|
stop func()
|
||||||
|
|
||||||
buf []rune
|
buf []rune
|
||||||
token string
|
token string
|
||||||
literal []rune
|
literal []rune
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -137,24 +118,7 @@ func (l *lexer) NextTok() bool {
|
||||||
case 'm':
|
case 'm':
|
||||||
l.loadString("mul")
|
l.loadString("mul")
|
||||||
l.token = "TokMUL"
|
l.token = "TokMUL"
|
||||||
if string(l.literal) != "mul" {
|
|
||||||
l.token = "TokILLEGAL"
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return true
|
return true
|
||||||
case 'd':
|
|
||||||
l.loadString("don't")
|
|
||||||
switch string(l.literal) {
|
|
||||||
case "don't":
|
|
||||||
l.token = "TokDONT"
|
|
||||||
case "do":
|
|
||||||
l.token = "TokDO"
|
|
||||||
default:
|
|
||||||
l.token = "TokILLEGAL"
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
|
|
||||||
case '(':
|
case '(':
|
||||||
l.loadRune("TokLPAREN")
|
l.loadRune("TokLPAREN")
|
||||||
return true
|
return true
|
||||||
|
@ -198,7 +162,7 @@ func readMul(lex *lexer) int {
|
||||||
if !lex.NextTok() || lex.token != "TokLPAREN" {
|
if !lex.NextTok() || lex.token != "TokLPAREN" {
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
if !lex.NextTok() || lex.token != "TokNUMBER" {
|
if !lex.NextTok() || lex.token != "TokNUMBER" {
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
@ -219,29 +183,3 @@ func readMul(lex *lexer) int {
|
||||||
fmt.Println(a, "*", b)
|
fmt.Println(a, "*", b)
|
||||||
return a * b
|
return a * b
|
||||||
}
|
}
|
||||||
|
|
||||||
func readDont(lex *lexer) bool {
|
|
||||||
if lex.token != "TokDONT" || string(lex.literal) != "don't" {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
if !lex.NextTok() || lex.token != "TokLPAREN" {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
if !lex.NextTok() || lex.token != "TokRPAREN" {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
func readDo(lex *lexer) bool {
|
|
||||||
if lex.token != "TokDO" || string(lex.literal) != "do" {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
if !lex.NextTok() || lex.token != "TokLPAREN" {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
if !lex.NextTok() || lex.token != "TokRPAREN" {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
|
@ -26,10 +26,9 @@ func TestExample(t *testing.T) {
|
||||||
t.Log(result)
|
t.Log(result)
|
||||||
is.Equal(result.valuePT1, 161)
|
is.Equal(result.valuePT1, 161)
|
||||||
|
|
||||||
result, err = run(bytes.NewReader(example2))
|
// result, err = run(bytes.NewReader(example2))
|
||||||
is.NoErr(err)
|
// is.NoErr(err)
|
||||||
|
// is.Equal(result.valuePT2, 48)
|
||||||
is.Equal(result.valuePT2, 48)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSolution(t *testing.T) {
|
func TestSolution(t *testing.T) {
|
||||||
|
@ -41,5 +40,5 @@ func TestSolution(t *testing.T) {
|
||||||
t.Log(result)
|
t.Log(result)
|
||||||
is.True(result.valuePT1 < 186942148) // first attempt too high
|
is.True(result.valuePT1 < 186942148) // first attempt too high
|
||||||
is.Equal(result.valuePT1, 184122457)
|
is.Equal(result.valuePT1, 184122457)
|
||||||
is.Equal(result.valuePT2, 0)
|
//is.Equal(result.valuePT2, 0)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user