-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
109 lines (95 loc) · 1.67 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package main
import (
"bufio"
"fmt"
"io"
"os"
"path"
"strings"
)
func input() *os.File {
input, err := os.Open(path.Join("2021", "10", "input.txt"))
if err != nil {
panic(err)
}
return input
}
func solve(r io.Reader) {
scanner := bufio.NewScanner(r)
score := 0
for scanner.Scan() {
line := scanner.Text()
if c, ok := isIllegal(line); !ok {
fmt.Printf("%s %c\n", line, c)
switch c {
case ')':
score += 3
case ']':
score += 57
case '}':
score += 1197
case '>':
score += 25137
}
}
}
if scanner.Err() != nil {
panic(scanner.Err())
}
fmt.Println(score)
}
func isIllegal(chunk string) (byte, bool) {
s := newStack(nil)
handleOpen := func(c byte) {
s.push(c)
//fmt.Println(s)
}
handleClose := func(c byte) bool {
//fmt.Println(s)
v := s.pop()
switch v {
case '[':
return c == ']'
case '{':
return c == '}'
case '<':
return c == '>'
case '(':
return c == ')'
default:
panic("bad char")
}
}
for i := 0; i < len(chunk); i++ {
c := chunk[i]
switch c {
case '[':
fallthrough
case '{':
fallthrough
case '<':
fallthrough
case '(':
handleOpen(c)
continue
case ']':
fallthrough
case '}':
fallthrough
case '>':
fallthrough
case ')':
ok := handleClose(c)
if !ok {
return c, false
}
default:
panic("bad char")
}
}
return 0, true
}
func main() {
solve(strings.NewReader("[({(<(())[]>[[{[]{<()<>>\n[(()[<>])]({[<{<<[]>>(\n{([(<{}[<>[]}>{[]{[(<()>\n(((({<>}<{<{<>}{[]{[]{}\n[[<[([]))<([[{}[[()]]]\n[{[{({}]{}}([{[{{{}}([]\n{<[[]]>}<{[{[{[]{()[[[]\n[<(<(<(<{}))><([]([]()\n<{([([[(<>()){}]>(<<{{\n<{([{{}}[<[[[<>{}]]]>[]]"))
solve(input())
}