|
|
@@ -0,0 +1,74 @@
|
|
|
+package main
|
|
|
+
|
|
|
+import (
|
|
|
+ "bufio"
|
|
|
+ "fmt"
|
|
|
+ "log"
|
|
|
+ "os"
|
|
|
+)
|
|
|
+
|
|
|
+func readLine(line string, arr []int) {
|
|
|
+ for i := range line {
|
|
|
+ if line[i] == '#' {
|
|
|
+ arr[i]++
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func readInput(file *os.File) ([][]int, [][]int) {
|
|
|
+ scanner := bufio.NewScanner(file)
|
|
|
+ var locks, keys [][]int
|
|
|
+
|
|
|
+ var isKey bool
|
|
|
+ var index int
|
|
|
+ arr := make([]int, 5)
|
|
|
+ for scanner.Scan() {
|
|
|
+ line := scanner.Text()
|
|
|
+ if line == "" {
|
|
|
+ if isKey {
|
|
|
+ keys = append(keys, arr)
|
|
|
+ } else {
|
|
|
+ locks = append(locks, arr)
|
|
|
+ }
|
|
|
+
|
|
|
+ arr = make([]int, 5)
|
|
|
+ index = 0
|
|
|
+ continue
|
|
|
+ }
|
|
|
+
|
|
|
+ if index == 0 {
|
|
|
+ if line[0] == '.' {
|
|
|
+ isKey = true
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if index != 0 && index != 6 {
|
|
|
+ readLine(line, arr)
|
|
|
+ }
|
|
|
+
|
|
|
+ index++
|
|
|
+ }
|
|
|
+
|
|
|
+ if isKey {
|
|
|
+ keys = append(keys, arr)
|
|
|
+ } else {
|
|
|
+ locks = append(locks, arr)
|
|
|
+ }
|
|
|
+
|
|
|
+ return locks, keys
|
|
|
+}
|
|
|
+
|
|
|
+func main() {
|
|
|
+ if len(os.Args) < 2 {
|
|
|
+ log.Fatal("You need to specify a file!")
|
|
|
+ }
|
|
|
+
|
|
|
+ filePath := os.Args[1]
|
|
|
+ file, err := os.Open(filePath)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatalf("Failed to open %s!\n", filePath)
|
|
|
+ }
|
|
|
+
|
|
|
+ locks, keys := readInput(file)
|
|
|
+ fmt.Println(locks, keys)
|
|
|
+}
|
