code.go 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. package main
  2. import (
  3. "bufio"
  4. "fmt"
  5. "log"
  6. "os"
  7. )
  8. func readInput(file *os.File) ([]string, int) {
  9. scanner := bufio.NewScanner(file)
  10. var maze []string
  11. start := -1
  12. for scanner.Scan() {
  13. line := scanner.Text()
  14. if line == "" {
  15. continue
  16. }
  17. if start < 0 {
  18. for i := range line {
  19. if line[i] == 'S' {
  20. start = i
  21. break
  22. }
  23. }
  24. }
  25. maze = append(maze, line)
  26. }
  27. return maze, start
  28. }
  29. func part1(maze []string, beams []int) int {
  30. var count int
  31. for row := 1; row < len(maze); row++ {
  32. for i := range beams {
  33. if beams[i] == 0 {
  34. continue
  35. }
  36. if maze[row][i] == '^' {
  37. if i > 0 {
  38. beams[i-1]++
  39. }
  40. if i < len(beams)-1 {
  41. beams[i+1]++
  42. }
  43. count++
  44. }
  45. }
  46. }
  47. return count
  48. }
  49. func main() {
  50. if len(os.Args) < 2 {
  51. log.Fatal("You need to specify a file!")
  52. }
  53. filePath := os.Args[1]
  54. file, err := os.Open(filePath)
  55. if err != nil {
  56. log.Fatalf("Failed to open %s!\n", filePath)
  57. }
  58. maze, start := readInput(file)
  59. beams := make([]int, len(maze[0]))
  60. beams[start] = 1
  61. fmt.Println("Part1:", part1(maze, beams))
  62. }