code.go 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  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. beams[i] = 0
  44. count++
  45. }
  46. }
  47. }
  48. return count
  49. }
  50. func main() {
  51. if len(os.Args) < 2 {
  52. log.Fatal("You need to specify a file!")
  53. }
  54. filePath := os.Args[1]
  55. file, err := os.Open(filePath)
  56. if err != nil {
  57. log.Fatalf("Failed to open %s!\n", filePath)
  58. }
  59. maze, start := readInput(file)
  60. beams := make([]int, len(maze[0]))
  61. beams[start] = 1
  62. fmt.Println("Part1:", part1(maze, beams))
  63. }